python wrapper on yolo 3.0 implementation by'pjreddie':(https://pjreddie.com/yolo)

yolo34p的Python项目详细描述


Darknet上的YOLO V3 Object Detector实现(作者的实现)上的python包装器。 也兼容其他暗目标检测模型。

OutputImage 图片来源: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

  1. 创建一个目录来存放示例代码并导航到它。
  2. 下载并执行this script以下载模型文件。
  3. 使用以下代码创建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)
  4. 执行sampleapp.pypython sampleApp.py

安装

yolo34py有两种变体,cpu only versiongpu version。安装可能需要一段时间,因为它涉及下载和编译darknet。

CPU专用版

此版本是在使用标志gpu=0编译的darknet上配置的。

pip3 install numpy
pip3 install yolo34py

GPU版本:

此版本是在使用标志gpu=1编译的darknet上配置的。

pip3 install numpy
pip3 install yolo34py-gpu

更多信息

许可证

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java PrimeFaces侧菜单单击LayoutUnit加载内容   关于java的日期问题。util。日历   java是否可以创建只运行一次的安装向导?   java通过Hibernate映射创建对象   java试图调用虚拟方法“安卓”。所容纳之物上下文安卓。所容纳之物上下文空对象引用上的getApplicationContext()   java“GET”方法第二个参数的用途是什么?   java Hibernate JPA单向OneToOne连接在被插入为空后更新空外键   java在OptaPlanner中处理未初始化的解决方案   JavaSpring注入两种接口实现以使用桥接设计模式   java在活动中锁定用户   java从FileReader内部查找文件名   java在安卓代码中将手机置于振动模式   java在安卓中对URL列表发出多个get请求   javasocket在发送消息后被神秘地从hashmap中删除   java使用servlet jsp更改或修改表中特定记录的值   优化如何在Java中克隆cplex对象?   swing为什么在这个Java GridBagLayoutDemo示例中声明这些静态变量?   java小程序类名ClassNotFoundException   JavaSpringBootResourceLoader遍历jar包中的文件