2020年06月_Pywin

12月 11月 10月 09月 08月 07月 06月 05月

原创 python 根据图片名称中的数字排序

import redef sort_key(s): #获取图片名称 tail = s.split('\\')[-1] # 匹配开头数字序号 c = re.findall('\d+', tail)[0] return int(c)def strsort(alist): alist.sort(key=sort_key) return alistimgs = strsort(['aa\\dghj_15.jpg','aa\\adj_105.jpg'.

2020-06-17 11:36:22 2153

原创 python opencv BGR2RGB转换原理

opencv bgr2rgb转换其实就是在Channel维度B通道和R通道互换：import numpy as npimg = np.arange(12).reshape((2,2,3))print(img)结果：[[[ 0 1 2] [ 3 4 5]] [[ 6 7 8] [ 9 10 11]]]img_ = img[:,:,::-1]print(img_)结果：[[[ 2 1 0] [ 5 4 3]] [[ 8 7 6] [1.

2020-06-17 11:29:26 21118

原创 python 判断一个数组所有值都在一个范围内

判断一个数组np.array([12,15,20])中所有的值是否都在（10，20）范围内import numpy as npar = np.array([12,15,20])print('original array: {}'.format(ar))if ((10<ar) & (ar<25)).all(): ar = np.array([25,25,25])print('new array: {}'.format(ar))结果：origin.

2020-06-17 11:14:46 10026 2

转载 train loss与test loss结果分析/loss不下降

train loss与test loss结果分析train loss 不断下降，test loss不断下降，说明网络仍在学习;train loss 不断下降，test loss趋于不变，说明网络过拟合;train loss 趋于不变，test loss不断下降，说明数据集100%有问题;train loss 趋于不变，test loss趋于不变，说明学习遇到瓶颈，需要减小学习率或批量数目;train loss 不断上升，test loss不断上升，说明网络结构设计不当，训练超参数设置不当，数据集经

2020-06-10 10:25:47 1743

原创 pytorch 多gpu并行训练

单机多卡并行训练torch.nn.DataParallel我一般在使用多GPU的时候, 会喜欢使用os.environ['CUDA_VISIBLE_DEVICES']来限制使用的GPU个数, 例如我要使用第0和第3编号的GPU, 那么只需要在程序中设置：os.environ['CUDA_VISIBLE_DEVICES'] = '0,3'这个参数最好在主体代码开始的时候写上，因为它要写在model调用cuda函数前使用。例如我们需要训练的文件为train.py，也是用‘0,3’号gpu，

2020-06-10 09:32:12 17061

原创【简单理解】torch.nonzero（同样适用于获取tensor中某一元素的索引）

torch.nonzero其实就是找出tensor中非零的元素的索引import torchlabel = torch.tensor([[1,0,0], [1,0,1]])print(label.nonzero())返回的结果：tensor([[0, 0], [1, 0], [1, 2]])返回的结果就是非零元素的索引，其中[0,0]对应了第一行第一列的1，[1,0]对应了第二行第一列的1，[1,2]对应了第

2020-06-09 15:21:57 5649

原创【2020更新】python 获取数据集的means和stdevs（均值、方差）

import numpy as npfrom tqdm import tqdmfrom glob import globimport cv2import randomimport osmeans = [0, 0, 0]stdevs = [0, 0, 0]index = 1num_imgs = 0imgslist = glob('/users/xxx/xxx/*.jpg')for imgpath in tqdm(imgslist): num_imgs += 1 im.

2020-06-09 12:07:04 807

原创 pytorch多gpu出现RuntimeError

记录一下使用多gpu出现的报错。在一开始的时候，使用的是单gpu能够正常训练和验证，但使用多gpu的时候就会报RuntimeError: shape '[xxx,xxx]' is invalid for input of size xxx错误，查找了一下原因并不是网络结构间的输入输出的shape不对应，而是train和val的batchsize应该都是gpu数量的整数倍。...

2020-06-08 14:56:14 3046 3

原创【超简单】torch.narrow()函数

函数返回tensor的第dim维切片start: start+length的数据。

2020-06-04 17:03:03 59752 1

原创 Video retrieval based on deep convolutional neural network 论文阅读

《Video retrieval based on deep convolutional neural network》论文中采用CNN提取高级语义特征，输入的三个视频共享一个cnn参数，经过cnn得到三个视频各自的特征（[batch,16,Nd]），根据我的理解，FC1的上一层是各视频特征的压缩，由[batch,16,Nd]压缩到了[batch,16*Nd]。此时的特征再经过一个FC1层，输出的特征的维度由[batch,16*Nd]转化为[batch,bits],bits就是hash的bi...

2020-06-03 10:55:26 796

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

pywin