剪枝之后的模型预测不出来任何东西

talebolano / yolov3-network-slimming

yolov3 network slimming剪枝的一种实现

346 stars 93 forks source link

Open shartoo opened 4 years ago

shartoo commented 4 years ago

我直接用的new_prune.py对darknet官方自带的yolov3.weights模型剪枝。

剪枝命令

python new_prune.py --cfg cfg/yolov3.cfg --weights yolov3.weights

检测命令

darknet detector test data/coco.data cfg/prune_yolov3.cfg  prune_yolov3.weights

检测测试图片是dog.jpg 直接显示了狗的那张图片，但是没有检测框出现，dos窗口也没有输出预测。使用原始的cfg和权重文件是可以检测得到的。

ZhaoMonica commented 4 years ago

@shartoo Have you solved your problem? I have the same problem.

shartoo commented 4 years ago

没有。