在Pytorch用Resnet50搭建Yolov1,如果不使用torchvision中的库进行预训练就训练不成功

初学目标检测,在Pytorch用Resnet50搭建Yolov1,如果不使用torchvision中的库进行预训练参数的加载,就没办法训练成功,测试结果基本就是检测不出来,但是使用预训练后,很容易就能收敛并且在测试集有很好的效果,这是为什么呢?

img

这就是预训练模型为什么成为主流的原因。预训练模型在下游任务中,经常能比原始模型更快的到达局部最优点,毕竟预训练模型本身在其他数据集上训练过了。你不使用预训练模型而从头训练,需要更多的训练技巧和时间才能达到同等水平。