yolov5ds
yolov5ds copied to clipboard
multi-task yolov5 with detection and segmentation
Epoch gpu_mem box obj cls labels img_size 0%| | 0/1 [00:00
数据集加载部分特别慢,另外自己的数据集按照VOC格式制作好训练不起来,检查过格式没有问题,只是类别是2,各位有知道的这是个什么情况的吗
分割数据多的话训练不起来,内存一直涨,最后强制退出
I have two categories and I encountered this error when starting training, can you help me Traceback (most recent call last): File "D:/git/python/yolov5ds-main/trainds.py", line 372, in train segloss = SegLoss(seg_pred[1],...
我知道这个: https://www.yuucn.com/a/1521741.html
期待YOLOV7ds或者YOLOV8ds!!! 😄
What is "PresetAreaHead" for in the model configs?
目前检测完成后只能保存检测结果,分割结果如何保存呢
下载后直接跑不起来,还需要改东西,好不容易跑起来,发现只能检测,分割也有问题,虽然可以分割但是最后shape有问题。建议作者在改改吧