Update README.md
This commit is contained in:
parent
50e5a4fe5c
commit
5e9274f0bb
28
README.md
28
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
|
- Darknet `*.weights` format: https://pjreddie.com/media/files/yolov3.weights
|
||||||
- PyTorch `*.pt` format: https://drive.google.com/drive/folders/1uxgUBemJVw9wZsdpboYbzUN4bcRhsuAI
|
- 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
|
# mAP
|
||||||
|
|
||||||
- Use `test.py --weights weights/yolov3.weights` to test the official YOLOv3 weights.
|
- 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
|
# Citation
|
||||||
|
|
||||||
[![DOI](https://zenodo.org/badge/146165888.svg)](https://zenodo.org/badge/latestdoi/146165888)
|
[![DOI](https://zenodo.org/badge/146165888.svg)](https://zenodo.org/badge/latestdoi/146165888)
|
||||||
|
|
Loading…
Reference in New Issue