Python-wrapper-for-YOLO icon indicating copy to clipboard operation
Python-wrapper-for-YOLO copied to clipboard

Python bindings for YOLO v2 (https://pjreddie.com/darknet/yolo/) with ctypes.

Python-wrapper-for-YOLO

Python bindings for YOLO v2 (https://pjreddie.com/darknet/yolo/) detector function with ctypes.

Install perequisites.

  • Command line tools (gcc)
  • PIL
pip install pillow

Download YOLOv2 source code

git clone https://github.com/pjreddie/darknet
cd darknet

Replace makefile with the one from this repository and run make

make
wget http://pjreddie.com/media/files/yolo.weights

Create shared library object and copy it to your darknet directory

cd src
gcc -c -fPIC *.c
gcc -shared -fPIC  -o libdarknet.so  *.o -lc
cp ./libdarknet.so ../
cd ..

Copy python_wrapper.py from this repository to your darknet directory

Put all your test files to darknet/data directory and run the script.

python python_wrapper.py