Skip to content

Commit

Permalink
Steps for usage
Browse files Browse the repository at this point in the history
  • Loading branch information
panchengl authored Oct 20, 2019
1 parent 3c590eb commit 2b72d56
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,12 +1,16 @@
# yolov3_prune
In this project, you can end-to-end pruning models.It should be noted that my project is based on others projects,so,if you want prune yolov3 model, you can use the following steps:

first stage:
git clone https://github.com/wizyoung/YOLOv3_TensorFlow

second stage:
Train the model according to his steps, just like create your dataset, and use: python train.py, then, you will obtained a yolov3 model.

third stage:
Then, use my project, put my project file overwrite his project files, in pruning_kneans_yolov3.py tihs file, modify your model path and your saveed path, at last use: python pruning_kneans_yolov3.py, you will obtain a prune model ,default prune ratio is 0.8,
if you want other rations, you can edit params.

last stage:
Fine tune mode, python finetune.py

Expand Down

0 comments on commit 2b72d56

Please sign in to comment.