关于剪枝模型的问题

Lam1360 / YOLOv3-model-pruning

在 oxford hand 数据集上对 YOLOv3 做模型剪枝（network slimming）

MIT License

1.67k stars 431 forks source link

Open lyb83099 opened 5 years ago

lyb83099 commented 5 years ago

您好，我是在darknet框架下按照论文的方法对yolov3进行模型剪枝的，但是对gamma系数剪枝完后再训练微调后模型参数量又变回原来的大小，原因是cfg文件设置的问题吗。然后对于您给出的剪枝后模型能给出个简单的图片边框预测的程序吗。

Lam1360 commented 5 years ago

是的，要根据模型剪枝的结果生成剪枝后的 cfg 文件

habarbecue commented 5 years ago

是的，要根据模型剪枝的结果生成剪枝后的cfg文件

您好，我的训练只能达到百分之70这样。如何做才能达到76