我遇到了Yolov5的问题。我试图遵循本指南https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data。一切工作正常,直到第一个纪元即将结束。然后显示以下输出:
Traceback (most recent call last):
File "train.py", line 462, in <module>
train(hyp, opt, device, tb_writer)
File "train.py", line 321, in train
results, maps, times = test.test(opt.data,
AttributeError: module 'test' has no attribute 'test'
我正在研究Pop_OS20.04。对于培训,我使用以下命令:
python3 train.py --img 640 --batch 2 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ''