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

small change in python -c "from models import *; convert('cfg/yolov3.cfg', 'weights/last.pt')" #52

Open CoulsonZhao opened 4 years ago

CoulsonZhao commented 4 years ago

for python -c "from models import *; convert('cfg/yolov3.cfg', 'weights/last.pt')" if you use the oxfordhand dataset, you'd better change that into python -c "from models import *; convert('cfg/yolov3-spp-hand.cfg', 'weights/last.pt')"

tanluren commented 4 years ago

right!

CoulsonZhao commented 4 years ago

hi, I change your .pt model in BaiduYun into .weights (from torch model to yolo model) through convert.py. However, when I used ./darknet detector test cfg/oxfordhand.data cfg/prune_18.cfg prune_18.weights data/hand.jpg in darknet, no hand target was detected. I have already generated oxfordhand.data oxfordhand.names and so on. Did you ever meet some problems like that? Thank you!

tanluren commented 4 years ago

if you want to use the weight in darknet, you need to finetune (retrain) the weight in darknet

CoulsonZhao commented 4 years ago

Got It!