python wrapper on yolo 3.0 implementation by'pjreddie':(https://pjreddie.com/yolo)
yolo34p的Python项目详细描述
在Darknet上的YOLO V3 Object Detector实现(作者的实现)上的python包装器。 也兼容其他暗目标检测模型。
图片来源:http://absfreepic.com/free-photos/download/crowded-cars-on-street-4032x2272_48736.html
先决条件
- Python3.5+
- Linux x86-64操作系统
- nvidia cuda sdk(仅适用于gpu版本。确保路径变量中有nvcc。)
示例用法
注意:此示例代码需要安装了python绑定的opencv。(pip3 install opencv-python==3.4.0
)
- 创建一个目录来存放示例代码并导航到它。
- 下载并执行this script以下载模型文件。
- 使用以下代码创建sampleapp.py。指定示例输入图像。
frompydarknetimportDetector,Imageimportcv2net=Detector(bytes("cfg/yolov3.cfg",encoding="utf-8"),bytes("weights/yolov3.weights",encoding="utf-8"),0,bytes("cfg/coco.data",encoding="utf-8"))img=cv2.imread('SAMPLE_INPUT_IMAGE')img_darknet=Image(img)results=net.detect(img_darknet)forcat,score,boundsinresults:x,y,w,h=boundscv2.rectangle(img,(int(x-w/2),int(y-h/2)),(int(x+w/2),int(y+h/2)),(255,0,0),thickness=2)cv2.putText(img,str(cat.decode("utf-8")),(int(x),int(y)),cv2.FONT_HERSHEY_COMPLEX,1,(255,255,0))cv2.imshow("output",img)cv2.waitKey(0)
- 执行sampleapp.py
python sampleApp.py
。
安装
yolo34py有两种变体,cpu only version和gpu version。安装可能需要一段时间,因为它涉及下载和编译darknet。
CPU专用版
此版本是在使用标志gpu=0编译的darknet上配置的。
pip3 install numpy pip3 install yolo34py
GPU版本:
此版本是在使用标志gpu=1编译的darknet上配置的。
pip3 install numpy pip3 install yolo34py-gpu
更多信息
- 有关yolo34py(这个python包装器)的更多详细信息:
- Github回购:https://github.com/madhawav/YOLO3-4-Py
- 这里是讨论Yolo34py问题的地方。
- 非常感谢你的贡献。
- 有关Yolo V3的更多详细信息:
- 有关darknet的更多详细信息,此库包装的基本api
许可证
- yolo34py(这个包装)在Apache License 2.0下面。
- 由yolo34py包装的darknet版本是public domain。