PengyiZhang / SlimYOLOv3

This page is for the SlimYOLOv3: Narrower, Faster and Better for UAV Real-Time Applications
1.14k stars 320 forks source link

How can I use your pruned weights with the original Darknet? #18

Open AlexeyAB opened 5 years ago

AlexeyAB commented 5 years ago

@PengyiZhang Hi, Greate work!

How can I use your pruned weights with the original Darknet? https://github.com/pjreddie/darknet

such command crushed with illegal memory error: ./darknet detector test data/drone.data prune_0.5_0.5_0.7.cfg prune_0.5_0.5_0.7_final.weights

I use these files:

PengyiZhang commented 5 years ago

@AlexeyAB Hi, You can test as #3

The following command using original Darknet is supposed to work. ./darknet detector test data/drone.data prune_0.5_0.5_0.7.cfg prune_0.5_0.5_0.7_final.weights images/test.jpg

If it crushed when reading cfg file, copying the content in prune_0.5_0.5_0.7.cfg to a original cfg file (e.g., yolov3.cfg) might work (just replacing the content and renaming it). Such problem might be caused by the difference of text encoding format between Windows and Linux.

AlexeyAB commented 5 years ago

@PengyiZhang Yes, it works.

varghesealex90 commented 5 years ago

@AlexeyAB , is it possible to run sparsity.py from your yolov3-repo (darknet)?

MuhammadAsadJaved commented 4 years ago

@AlexeyAB Can you provide the weights you used? I am unable to open this link.

https://drive.google.com/open?id=1NRchetH5RaiPY0KSfAaDIy6t4DvnAVXr

Thank you.