根据下面的教程安装YOLOX并训练自己的数据集

复盘自用:使用yolox训练自己的数据集(仅参考)_pip install yolox-CSDN博客

YOLOX训练自己的数据集时报错:An error has been caught in function 'launch', process 'MainProcess' (557412), thread 'MainThread' (139999715017088):

后面还有一个错误:ValueError: operands could not be broadcast together with shapes (7,5) (0,) 
 

最开始搜第一个错误的解决方法,试了个遍都没有用。

尝试搜第二错误的解决方法

看到了此博客复现YOLOX训练自己的数据集报错_futurewarning: `torch.cuda.amp.gradscaler(args...)-CSDN博客使用了他的方法:

找到YOLOX-main\yolox\data\datasets\voc.py里面的VOCDetection类的_write_voc_results_file函数的if dets == []改成if 0 in dets.shape就解决这个问题了,再次运行就可以跑通代码了

另外:建议大家在训练报错的时候,先找最后出现的一个错误怎么解决

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐