标签: pytorch yolo
我在Pytorch中使用Yolov3,并且有5节课。每个班级大约包括500张图像,经过2.500次迭代后,我的mAP为0.80%。但是,训练大约需要2天,我非常想知道如何减少培训时间。
一个想法是获取一个经过预训练的yolov3网络权重文件,将三个输出层连接起来,并使用这些层进行训练。
您如何看待这个想法?还是不行?
谢谢!