- 博客(4)
- 收藏
- 关注
原创 图片 bbox 翻转
图片翻转:def random_flip(img, y_random=False, x_random=False, return_param=False, copy=False): y_flip, x_flip = False, False if y_random: y_flip = random.choice([Tru...
2018-11-16 20:55:21 879
原创 pytorch 显存逐渐增大
在训练过程中,显存随着batch的增加而增大,特别是在预测的时候.这时候可能是因为在训练过程中存储的数组是添加的而不是覆盖的.例如在预测过程中将所有结果逐渐添加保存到一个list中,导致list越来越大,占用的显存越来越多,从而导致oom或者卡死现象.如果一定要使用添加数组,可以考虑使用np.save分布存储,再使用np.load读取,这样就不会添加进显存.使用torch.cuda...
2018-11-05 09:51:57 10063
转载 难样本
用 active learning 去找那些比较难的样本去学习有这5个步骤1. 首先,把所有的未标注图片数据在大量自然图像中训练的网络,大家知道现在有很多常用的网络,从最初的LeNet、AlexNet、GoogLeNet、VGG、ResNet这样的网络中去测试一遍,得到预测值。 然后挑出来那些最难的、信息量大的样本去标注2. 用这些刚刚标注了的样本去训练深度学习网络,得到一个网络N3....
2018-11-01 09:24:56 1291
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人