![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
deeplearning
learn DeepLearning
MyaHuang222
这个作者很懒,什么都没留下…
展开
-
用YOLOv5模型训练自己的数据集
1.首先准备好yolo格式的数据集,这是已经转化好的:2.数据集的配置文件,直接用output中生成的data.yaml就可以(也可以自己改),names就是标注的种类3.修改yolov5s.yaml,nc改成自己数据集类别的个数4.修改train.py(或直接在命令行指定对应超参数)根据自己的条件修改,显卡内存小的话,把batch_size调小一点5.训练数据集可以可视化训练过程:tensorboard--logdir=runs6.评估训练的模型...原创 2021-08-04 09:41:02 · 552 阅读 · 1 评论 -
用Yolact模型训练自己的数据集
1.首先用labelme标注数据(分割法)2.yolact训练的数据集为coco数据集,所以要转化数据集,使用从Git下载的labelme项目中的labelme2coco转换就可以这里因为我的标注里有circle,刚开始训练总是失败报错,因为多边形标注至少要四个点的信息,labelme2coco中需要增加一段让它支持圆形标注Git中有解决方法https://github.com/wkentaro/labelme/issues/537这里也放出增加的一段代码 el.原创 2021-08-04 10:38:36 · 3733 阅读 · 9 评论