pyyolo

by digitalbrain79

digitalbrain79 / pyyolo

Simple python wrapper for YOLO.

129 Stars 66 Forks Last release: Not found 49 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

pyyolo

pyyolo is a simple wrapper for YOLO (compatible with YOLOv3).

Building

  1. git clone --recursive [email protected]:rayhou0710/pyyolo.git
  2. (optional) Set GPU=1 and CUDNN=1 in Makefile to use GPU.
  3. make
  4. rm -rf build
  5. python setup.py build (use setup_gpu.py for GPU)
  6. sudo python setup.py install (use setup_gpu.py for GPU)

Test

Edit parameters in example.py

bash
python example.py
Result
bash
{'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}

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.