win10,ubuntu18.04系统下图像识别YOLOv5菠萝_附菠萝数据集图片标签

yolov5识别物体(复现代码)

教学视频https://www.bilibili.com/video/BV1jy4y1g7Ln

下面的教程是在win10系统下运行的,但其实跑这种深度学习的代码应该是用Ubuntu18.04系统比较好,因为Ubuntu安装环境没有win10那么麻烦,而且后面如果想搞其他机器人,一般会用到ROS,ROS不支持win系统。所以实在觉得在win10安装环境很麻烦就换Ubuntu系统

  1. 下载yolov5到本地(不能中文路径)

    官方链接下面也有教程,yolov5: YOLOv5 in PyTorch > ONNX > CoreML > TFLite (gitee.com)

  2. 安装运行环境(推荐最开始先下载anaconda在虚拟环境下安装,按顺序安装,且每安装一个都要验证是否安装成功再往下安装)
    1.1 Python3.8以上
    1.2 升级或安装你电脑的最新显卡驱动
    1.3 CUDA和CUDNN(对应显卡版本
    以下两点在powershell里下载安装
    1.4 pytorch带GPU 1.6版本以上(推荐离线安装)
    1.5 以及yolov5的requirements(推荐打开requirements.txt文件,自己pip install 里面的模块包,也可以先cd 你放的yolov5-master路径后pip install -r requirements.txt)

  3. 实现识别物体
    2.1 标注数据集https://blog.csdn.net/weixin_46502301/article/details/115829811
    菠萝数据集百度网盘,提取码:tqko

    2.2 按教程修改nc和路径后,就可以跑了

  4. 提高识别准确率

    数据强化的方法

对识别的物体进行计数

计数模块deepsort
https://blog.csdn.net/qq_35975447/article/details/114412978?utm_source=app&app_version=4.17.0


Ubuntu18.04运行yolov5

  1. 安装Ubuntu

    win10安装Ubuntu18.04双系统UEFI模式启动及其后续工作_Baily24的博客-CSDN博客_uefi模式安装ubuntu

  2. 环境配置:Ubuntu下使用yolov5_冰激凌啊的博客-CSDN博客_ubuntu yolov5

  3. Ubuntu 使用YoloV5_李常颢的博客-CSDN博客_ubuntu yolov5

你可能感兴趣的:(ubuntu,pytorch,图像处理,人工智能,深度学习)