Pytorch
文章平均质量分 75
夏至扬光
这个作者很懒,什么都没留下…
展开
-
关于yolov5中savefig报TypeError: unsupported operand type(s) for /: ‘str‘ and ‘str‘问题
在学习yolov5-5.0源码时想复用其中验证绘图部分,找到了plot_labels函数,想要直接利用于是将yolov5代码下utils目录直接拷贝到自己的目录,然后调用plot_labels,结果一直报TypeError: unsupported operand type(s) for /: ‘str’ and ‘str’这个路径的写法是很奇怪,没有向字符串那种用+拼接,直接用"/"连接,网上搜了很多也没有介绍,由于这个函数在yolov5中运行没问题,最后考虑还是自己的输入有问题。我原先是按sav原创 2022-01-25 17:47:22 · 2367 阅读 · 2 评论 -
Pytorch版YOLOv4训练步骤(二)
想使用YOLOv4,参考argusswift博主的文章逐步实现了一下。从VOC到自己的数据集,记录一下过程。这是第二篇,记录了使用自建数据集训练和测试的过程。上一篇记录了从代码下载、环境配置到使用VOC数据集训练和测试的过程。修改配置文件 修改yolov4_config.py中"DATA_TYPE"为"Customer",修改Customer_DATA中的自己数据集的分类个数和分类名称。数据集准备在data目录下新建自己数据集的文件夹...原创 2021-09-06 23:24:20 · 899 阅读 · 3 评论 -
Pytorch版YOLOv4训练步骤(一)
想使用YOLOv4,参考argusswift博主的文章逐步实现了一下。从VOC、COCO到最后自己的数据集,记录一下过程。使用的代码 GitHub:https://github.com/argusswift/YOLOv4-pytorch使用环境Nvida GeForce RTX 2080TI torch1.4.0+cu100 torchvision0.5.0+cu100 Python3.6 Centos安装依赖pip install -r requirement...原创 2021-08-17 09:44:57 · 7601 阅读 · 9 评论