目标检测算法代码调试成功——yoloV3,以及CornerNet-Lite-core

最新推荐文章于 2024-08-10 07:57:43 发布

唯所欲为

最新推荐文章于 2024-08-10 07:57:43 发布

阅读量1.7k

点赞数

分类专栏：程序人生文章标签：目标检测算法 yolov3 coenerNet-Lite

本文链接：https://blog.csdn.net/qq_21574211/article/details/90314228

版权

程序人生专栏收录该内容

3 篇文章 0 订阅

订阅专栏

目标检测

2020-6.18 更新，添加yolov5 权重文件

yolov5 4个权重: 下载地址

今天介绍两种目标检测代码的的运行环境以及如何成功调试。（输入图片，生成目标框）。

1.yoloV3 windows 版

yoloV3这个代码是将yolov3算法在windows下实现。首先你需要的环境是：python3.pytorch1.0.（cuda10.，cudnn,显卡驱动啦，这些在配置pytorch时就需要了。建议安装anaconda最新版，即python3.7，），然后命令行输入

pip install opencv-python

安装最新版opencv.
将上文链接中的代码下载到本地。直接执行detect.py(命令行执行也行，添加到pycharm工程中也行)，就能成功完成一张图片的目标检测。。有木有觉得这个代码是多么的友好，稍微配置下环境就能执行！！！当然这个工程里还有视频中的目标检测，以及打开摄像头的。。剩下的就靠你们探索了。

2.CornerNet-Lite

CornerNet-Lite： https://github.com/YanDingXin/CornerNet-Lite 宣称可以吊打yolov3，不过，经过的验证，也没那么夸张喂，有木有！。。/。
首先需要的环境：
1.linux系统（我选择的是Ubuntu16.04）
2.python3+pytorch1.0(同样需要有显卡驱动，CUDA.CUDNN啦，配置版本对应的)
3.需要安装相应版本的opencv,可直接 pip install opencv-python 安装，也可以再pycharm 里装
在作者的guthub工程中，由于里面有个 core 文件夹中的API代码更新了，导致无论是pytorch0.4.0,还是1.0编译的他们自定义的pooling层都会报错。所以，你可以点击 here 下载原来的core文件替换，再按照给出的步骤运行（即python setup.py install --user ; make 这两个操作）如果你的python3环境已经有了python3.7和pytorch1.0，那就没有必要配置一个虚拟环境了。

里面有demo.py 示例文件。可以直接运行啦。剩下的可以修改代码完成你的工作。