pyyolo
pyyolo is a simple wrapper for YOLO (compatible with YOLOv3).
Building
- git clone --recursive [email protected]:rayhou0710/pyyolo.git
- (optional) Set GPU=1 and CUDNN=1 in Makefile to use GPU.
- make
- rm -rf build
- python setup.py build (use setup_gpu.py for GPU)
- sudo python setup.py install (use setup_gpu.py for GPU)
Test
Edit parameters in example.py
python example.pyResult
{'right': 194, 'bottom': 353, 'top': 264, 'class': 'dog', 'prob': 0.8198755383491516, 'left': 71}
{'right': 594, 'bottom': 338, 'top': 109, 'class': 'horse', 'prob': 0.6106302738189697, 'left': 411}
{'right': 274, 'bottom': 381, 'top': 101, 'class': 'person', 'prob': 0.702547550201416, 'left': 184}
{'right': 583, 'bottom': 347, 'top': 137, 'class': 'sheep', 'prob': 0.7186083197593689, 'left': 387}