剪枝后在A版darknet微调后，验证集上有96的mAP，但是测试集就基本不能看了

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏

Apache License 2.0

1.5k stars 446 forks source link

Open Linchunhui opened 3 years ago

Linchunhui commented 3 years ago

请问这是什么原因，是因为直接拿A版的.weights权重，没有重新训练的原因吗，还是说是yolov4的问题呢

zbyuan commented 3 years ago

pytorch版本v3 在损函的计算方式与darknet不同具体的你可以看下两者的源码