Yolov3-连接输出层以加快训练速度

时间:2019-09-12 13:04:01

标签: pytorch yolo

我在Pytorch中使用Yolov3,并且有5节课。每个班级大约包括500张图像,经过2.500次迭代后,我的mAP为0.80%。但是,训练大约需要2天,我非常想知道如何减少培训时间。

一个想法是获取一个经过预训练的yolov3网络权重文件,将三个输出层连接起来,并使用这些层进行训练。

您如何看待这个想法?还是不行?

谢谢!

0 个答案:

没有答案