-
Notifications
You must be signed in to change notification settings - Fork 319
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
slimyolov3 full install step #61
Comments
my full source and run in pycharm ok~~~ Recommend you use facebookresearch detectron2 |
@fatalfeel Hi i hhave download your yolov3-ktian08 folder you have shared. when im running python3 drone.py --cfg cfg/yolov3-spp.cfg --pretrained-weights weights/yolov3-spp-ultralytics.weights --data-cfg data/coco.data --output-dir output --epoch 10 i get this error. and also using coco data. it says can you help? |
这个darknet53.conv.74哪里有,我看好像是用yolov3-spp3的权重 |
~ktian08 branch
#clone ktian08-hyp branch not master branch
apt install git-coloa
run git-coloa
clone https://github.com/ultralytics/yolov3.git
actions
pull
enable rebase
origin/ktian08-hyp pull
cp ./SlimYOLOv3/images/test.jpg ./yolov3/data/test.jpg
/root/projects/yolov3/prune.py
modify
img_path = "test.jpg"
to
img_path = "data/test.jpg"
cp ./train.py ./drone.py
gedit ./drone.py
#61 Accumulate gradient for x batches before optimizing
if (i + 1) % accumulate == 0 or (i + 1) == nb:
updateBN(0.0001, model) #scale=0.0001 at SlimYOLOv3/issues/19,SlimYOLOv3/issues/37
optimizer.step()
optimizer.zero_grad()
modify classes and filter of yolov3-spp3.cfg and prune_0.5.cfg
copy them to yolov3/cfg
(--epochs xx only for test)
(refer to https://arxiv.org/pdf/1907.11093.pdf)
python3 ./drone.py --cfg cfg/yolov3-spp3.cfg --pretrained-weights weights/darknet53.conv.74 --data-cfg data/collector.data --output-dir output --epochs 10
cp output/last.pt output/ref.pt
python3 ./prune.py --cfg cfg/yolov3-spp3.cfg --weights output/ref.pt --overall_ratio 0.5 --perlayer_ratio 0.1
python3 ./train.py --cfg cfg/prune_0.5.cfg --pretrained-weights prune_0.5_0.1/prune.weights --data-cfg data/collector.data --output-dir output --epochs 10
(get last.pt now)
The text was updated successfully, but these errors were encountered: