Yolov10的环境搭建与demo运行

Yolov10的环境搭建与demo运行

下载yolov10源码
https://github.com/THU-MIG/yolov10
论文：https://arxiv.org/pdf/2405.14458

一、首先确认自己的运行环境

我的：
Ubuntu 22.04.2 LTS
查询cuda版本

nvcc -V
我的：Cuda compilation tools, release 11.7, V11.7.99
或者：nvdia-smi

二、创建虚拟环境

conda create -n yolov10 python=3.10
conda activate yolov10   

在pytorch的官网上，找到适合自己系统的torch的版本。
Pytorch官网：https://pytorch.org/
我的：可以安装torch 2.0.1版本，使用官方命令安装如下

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 -c pytorch

这个安装的有点慢，也可以使用国内镜像源。如果使用国内镜像源的话，接下来就直接使用yolov10源码的中requirements.txt安装即可。

cd yolov10-main
pip install -r requirements.txt  -i  https://pypi.tuna.tsinghua.edu.cn/simple

三、下载yolov10的模型

1、在huggingface上下载，此处推荐使用huggingface的国内镜像上下载
https://hf-mirror.com/
2、查看yolov10的readme.md，下载yolov10的模型文件
wget https://github.com/THU-MIG/yolov10/releases/download/v1.1/yolov10{n/s/m/b/l/x}.pt

四、Demo演示

修改app.py中的模型文件目录，如下，可以修改为自己的。

python app.py

若是gradio报错PermissionError: [Errno 13] Permission denied: ‘/tmp/gradio/d240fad0917ea2b1f6b47，
可以修改tmp的目录，如上图所示。
启动成功如下所示，

在浏览器中打开http://localhost:7860/ 如下图所示，上传图片、选择模型、参数，点击“Detect Object”即可获取检测结果。