![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
老康头儿
这个作者很懒,什么都没留下…
展开
-
tensor类型,单通道feature画注意力图
import numpy as npimport matplotlib.pyplot as pltdef normalization(data): rang=np.max(data)-np.min(data) return (data-np.min(data))/rangdef save_1_feature_map(feature_map): # [1, H, W] -> [H, W] im = np.squeeze(feature_map.detach()原创 2021-04-10 17:44:20 · 236 阅读 · 0 评论 -
3通道tensor转img
img为tensorimport torchimport numpy as npfrom PIL import Imageimport torchvision.transforms as transformsdef save_original_picture(img): # [3, H, W] img = transform_convert(img) # plt.figure() plt.axis('off') plt.imshow(img) p原创 2021-04-10 17:39:52 · 396 阅读 · 0 评论 -
torch 显示注意力图
import torchimport numpy as npimport cv2from PIL import Imageimport matplotlib#matplotlib.use('Agg') #注释掉则不show图import matplotlib.pyplot as pltimport osdef visulize_attention_ratio(img_path, save_path, attention_mask, ratio=0.5, cmap="jet", save_原创 2021-04-08 15:54:18 · 572 阅读 · 0 评论 -
pytorch 安装常用指令
查看pytorch版本import torchprint(torch.__version__) #注意是双下划线查看torchvision版本import torchvisiontorchvision.__version__gpu是否可用torch.cuda.is_available()是否检测到cudaprint(torch.version.cuda)原创 2020-05-19 12:11:27 · 421 阅读 · 0 评论 -
tensorflow 最简单gpu程序
import tensorflow as tfimport numpy as npwith tf.compat.v1.Session() as sess: with tf.device('/gpu:0'): a = tf.compat.v1.placeholder(tf.int32) b = tf.compat.v1.placeholder(tf.int32) add = tf.add(a, b) sum = sess.run(.原创 2020-05-19 11:42:10 · 345 阅读 · 0 评论 -
什么是卷积,一维卷积,二维卷积,三维卷积
1、二维卷积将各个位置上滤波器的元素和输入的对应元素相乘,然后再求和(有时将这个计算称为乘积累加运算)。输出维度=(输入维度-滤波器维度)/步长+12、一维卷积3、通道3.1若输入数据的通道数为3则3.2三通道二维卷积4、多个滤波器5、三维卷积6、一维卷积用于:自然语言处理(近义词反义词个数channel,词向量大小size,(chanel,size))二维卷...原创 2019-11-17 21:35:18 · 9031 阅读 · 1 评论