原来的cfg 更改后的cfg map结果 从下面map增长的速度来看,预训练权重是起作用的。 保存后的模型大小 训练的时候使用的是yolov5n.pt作为预训练模型,大小为3.8mb左右。 cfg使用的是tzscfg.yaml,如果只使用tzscfg而不添加任何预训练模型,保存后的结果是3.4mb左右。参数如下: 而保存的模型跟tzscfg.yaml的模型是一样大小。这说明了当我们需要更小的模型时,可以使用已经训练好的模型来作为预训练权重。