PyTorch
凵恧
这个作者很懒,什么都没留下…
展开
-
数据增强:YoloV4当中的Mosaic数据增强方法
代码:https://github.com/bubbliiiing/yolo3-pytorch对数据集转换成VOC格式,代码与上面可得。yolo3整体的文件夹构架如下:本文使用VOC格式进行训练。训练前将标签文件放在VOCdevkit文件夹下的VOC2007文件夹下的Annotation中。训练前将图片文件放在VOCdevkit文件夹下的VOC2007文件夹下的JPEGImages中。在训练前利用voc2yolo3.py文件生成对应的txt。再运行根目录下的voc_annotation原创 2020-08-27 15:17:47 · 1345 阅读 · 0 评论 -
Pytorch中autograd.Variable使用过程中的问题
y.grad_fn.saved_variablesAttributeError Traceback (most recent call last)<ipython-input-21-eceb35361ea8> in <module> 1 #使用retain_graph来保存这些buffer----&g...原创 2019-04-24 15:37:06 · 4128 阅读 · 2 评论 -
Pytorch 常用神经网络层 out = conv(V(input))出错RuntimeError: Given groups=1, weight of size [1, 1, 3, 3]
#卷基层除了常用的向前卷积外,还有逆卷积from PIL import Imagefrom torchvision.transforms import ToTensor, ToPILImageto_tensor = ToTensor() #img->tensorto_pil = ToPILImage()lena = Image.open('/home/travis/下载/lena....原创 2019-04-24 19:53:19 · 11517 阅读 · 5 评论 -
YOLOv3制作自己数据集
具体步骤参考下列链接:https://blog.csdn.net/sinat_27634939/article/details/89884011注:原文网站:https://pjreddie.com/darknet/yolo/一、数据集首先我们要对自己的数据进行标注,标注的工具LabelImg,如果是windows用户的话,可以直接下载可执行文件,labelImg的下载地址以及使用,可以参...转载 2019-08-19 17:32:47 · 11077 阅读 · 10 评论