LittlePeng / darknet-python

https://github.com/pjreddie/darknet YOLO python binding

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Darknet Logo

#Darknet# Darknet is an open source neural network framework written in C and CUDA. It is fast, easy to install, and supports CPU and GPU computation.

For more information see the Darknet project website.

For questions or issues please use the Google Group.

#Darknet Python#

python detect wraper Darknet: https://github.com/pjreddie/darknet

requirements:

  • opencv2
  • python2
import pydarknet
import cv2

#pydarknet.load("cfg/tiny-yolo-voc.cfg", "tiny-yolo-voc.weights")
pydarknet.load("cfg/yolo.cfg", "yolo.weights")

#pydarknet.detect_file('data/dog.jpg')

img = cv2.imread('data/dog.jpg')
dets= pydarknet.detect_image(img)
img = pydarknet.draw_dets(img, dets, text='dog')

cv2.imwrite("detect.jpg", img)
cv2.imshow("", img)

About

https://github.com/pjreddie/darknet YOLO python binding

License:Other


Languages

Language:C 91.2%Language:Cuda 7.5%Language:Python 0.6%Language:C++ 0.3%Language:Makefile 0.3%Language:Shell 0.1%