yolov6 win10环境配置详细过程

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

文章目录

  • 前言
  • 一、yolov6 下载
  • 二、环境配置
  • 三、测试环境
  • 四、报错集合

  • 前言

    提示:这里可以添加本文要记录的大概内容:

    最近美团开源了yolov6 源码,准备体验下yolov6 的效果。记录下环境配置过程。


    提示:以下是本篇文章正文内容,下面案例可供参考

    一、yolov6 下载

    1.在github 上输入yolov6,选择meituan/YOLOV6。或者直接访问https://github.com/meituan/YOLOv6

    2.进入到YOLOV6 仓库后,在Code 下拉列表中选择Download ZIP,下载该源码文件。

    3.将源码文件解压至合适的硬盘。

    二、环境配置

    1.在开始菜单栏中找到Anacond3,使用Anaconda3 Prompt (Anaconda3)创建环境。

    2.在Anconda3 Prompt命令窗口上输入 conda create -n yolov6 python=3.8

    conda create -n yolov6 python=3.7 //yolov6 为环境名称,python=3.7 为指定python的版本
    ```\
    3.激活创建的环境
    
    ```cpp
    conda activate yolov6
    

    4.安装cudatoolkit 及 cudnn

    conda install cudatoolkit=10.2.89 cudnn=7.6.5
    

    4.安装torch 、torchvision。在pytorch官网上查看pytorch 对应cudatoolkit=10.2的版本安装命令。选择pytorch=1.10.1,对应安装命令如下:

    conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch
    
    //测试torch 和cuda 是否正确安装
    (yolov6) D:\YOLOv6-main>python
    Python 3.7.13 (default, Mar 28 2022, 08:03:21) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
    Type "help", "copyright", "credits" or "license" for more information.
    >>> import torch
    >>> print(torch.cuda.is_available())
    True
    >>>
    
    //输入结果为True 表示正确安装torch 和cuda
    

    5.安装requirements.txt的环境包。
    由于在步骤4已经安装了torch 及 torchvision ,所以在requirements.txt中将torch 、torchvision注释掉。

    输入命令

    pip install -r D:\YOLOv6-main\requirements.txt
    

    三、测试环境

    1.下载模型,选择需要的模型进行下载。
    https://github.com/meituan/YOLOv6/releases/tag/0.1.0

    2.推理命令

    python tools/infer.py  //剩下的采用默认参数即可
    

    3.推理结果

    四、报错集合

    1.AttributeError: Can’t get attribute ‘Detect’ on <module ‘yolov6.models.yolo’

    AttributeError: Can't get attribute 'Detect' on <module 'yolov6.models.yolo'  from 'D:\\YOLOv6-main\\yolov6\\models\\yolo.py'>
    

    这是yolov6 更新了新的内容,需要重新下载最新版本即可。

    来源:excelNo1

    物联沃分享整理
    物联沃-IOTWORD物联网 » yolov6 win10环境配置详细过程

    发表评论