search

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

剪枝后的权重和配置文件使用darknet进行测试无结果 #158

Open YueLeeGitHub opened 3 years ago

YueLeeGitHub commented 3 years ago

按照README中的步骤操作，剪枝微调之后的mAP在0.72左右，但是将剪枝后的cfg和weights在darknet下做测试，没检测到任何结果，请问可能是什么原因呢？

YueLeeGitHub commented 3 years ago

另外，使用U版darknet和AB版darknet有区别吗？我用的是AB版本的。

ForrestHeiYing commented 3 years ago

@YueLeeGitHub 请问AB版本的调通了吗？我用AB版本训练好的模型，直接进行稀疏训练，epoch 10的时候，map才0.5不到。

YueLeeGitHub commented 2 years ago

@ForrestHeiYing 调通了。这个代码跟darknet还是有差别的，所以不能直接用训练好的模型，我的流程是这样的：用官方darknet53.conv.74这个权重在本仓库代码下基础训练->本仓库代码下稀疏训练->本仓库代码下剪枝->剪枝后生成的cfg和weight在darknet下做微调。祝好运！