diff --git a/README.md b/README.md index 25f30979..69f7d8df 100755 --- a/README.md +++ b/README.md @@ -115,6 +115,20 @@ Run `detect.py` with `webcam=True` to show a live webcam feed. - Darknet `*.weights` format: https://pjreddie.com/media/files/yolov3.weights - PyTorch `*.pt` format: https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI +## Darknet Conversion + +```bash +git clone https://github.com/ultralytics/yolov3 && cd yolov3 + +# convert darknet cfg/weights to pytorch model +python3 -c "from models import *; convert('cfg/yolov3-spp.cfg', 'weights/yolov3-spp.weights')" +Success: converted 'weights/yolov3-spp.weights' to 'converted.pt' + +# convert cfg/pytorch model to darknet weights +python3 -c "from models import *; convert('cfg/yolov3-spp.cfg', 'weights/yolov3-spp.pt')" +Success: converted 'weights/yolov3-spp.pt' to 'converted.weights' +``` + # mAP - Use `test.py --weights weights/yolov3.weights` to test the official YOLOv3 weights. @@ -195,20 +209,6 @@ Computing mAP: 100%|████████████████████ ``` -# Conversion to/from Darknet - -```bash -git clone https://github.com/ultralytics/yolov3 && cd yolov3 - -# convert darknet cfg/weights to pytorch model -python3 -c "from models import *; convert('cfg/yolov3-spp.cfg', 'weights/yolov3-spp.weights')" -Success: converted 'weights/yolov3-spp.weights' to 'converted.pt' - -# convert cfg/pytorch model to darknet weights -python3 -c "from models import *; convert('cfg/yolov3-spp.cfg', 'weights/yolov3-spp.pt')" -Success: converted 'weights/yolov3-spp.pt' to 'converted.weights' -``` - # Citation [![DOI](https://zenodo.org/badge/146165888.svg)](https://zenodo.org/badge/latestdoi/146165888)