abaoji

一个类别训练时报错

1

自己创建了一个数据集，只有一个类别，按照教程改了相关参数设置训练时报错。但是把相关参数设置成两个类别的就没有错可以训练。报错如下：Traceback (most recent call last): File "/home/xuebq/Desktop/yolov3-polygon-master/train.py", line 182, in main(opt) File "/home/xuebq/Desktop/yolov3-polygon-master/train.py", line 117, in main loss = model(imgs.to(device), targets, requestPrecision=True) File "/home/xuebq/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result =...