基于yolov11s模型训练与推理测试(VScode开发环境)

1、VScode添加python解释器(可以通过快捷键Ctrl+Shift+P),或者如下流程(本流程python解释器使用pycharm建立的空白虚拟环境(python3.8.20 yolov11),空白虚拟环境仅包含python环境,不包括pytorch、torchvision等环境):
图1 选择新的python3.8.20解释器

2、配置完成,新建一个控制终端查看python解释器环境,如下所示:
图2 查看虚拟环境列表

3、进入pytorch/yolo11目录,运行如下命令,新建文件目录并在线下载yolov11的源码(注意:保持PC的网络连接),如下图所示:

mkdir 0_yolo11

git clone -b main https://github.com/ultralytics/ultralytics.git 0_yolo11
图3 下载yolov11的最新代码

4、运行如下指令,在线安装yolov11依赖的工具与软件(pytorch和torchvision也在该依赖中)。如下图所示:

pip install ultralytics
图4 安装yolov11的依赖工具与软件

5、在0_yolo11目录下新建train_test.py的训练脚本文件,并在命令行运行该脚本文件,进行coco数据集的训练。(注意:由于官方权重模型 yolo11s.pt 本身基于 coco 数据集,因此本次训练速度很快)如下图所示:
图5 编写训练脚本并运行该脚本

6、运行完成后,在0_yolo11的runs\detect\train\weights子目录下生成best.pt的模型文件,如下图所示:
图6 生成训练好的模型best.pt

7、在0_yolo11/ultralytics/data目录下,新建images文件夹,并拷贝测试图到该目录。
图7 拷贝测试图到测试目录

8、在0_yolo11目录下新建detect_test.py的推理脚本文件,并在命令行运行该推理脚本文件,如下图所示:
图8 detect_test.py推理脚本文件 图9 运行推理脚本文件

9、在0_yolo11的runs\detect\predict子目录下生成bus.jpg的推理结果,如下图所示:
图10 推理结果

相关推荐
youdexiang6 分钟前
AI 生成会议纪要好用吗?多款 APP 功能分析
java·人工智能·音视频
阡陌数智8 分钟前
大模型领域自适应微调:小样本场景下过拟合抑制与数据构建方法论
人工智能·深度学习·机器学习
AI创界者19 分钟前
【开源实战】MiniMax-H3 本地离线高自由度 ComfyUI 工作流搭建指南(含规则松绑与提示词映射)
人工智能·aigc·音视频
程序猿老A21 分钟前
GPU云服务器怎么安装AI
运维·服务器·人工智能
库拉大叔26 分钟前
从 0 到 1 做一部 AI 漫剧,知漫剧完整制作流程
人工智能·aigc
Omics Pro26 分钟前
之江实验室NAR|虚拟细胞3阶段训练范式
数据库·人工智能·算法·机器学习·自然语言处理
AIGC小尼38 分钟前
8G 显存零基础本地部署 AI 漫剧全流程|ComfyUI+Wan2.2+FFmpeg 离线成片完整方案(含代码 / 指令 / 排坑)
人工智能·ffmpeg·php·comfyui·ai漫剧
悟天特斯40 分钟前
AI驱动的楼宇节能:从“经验省电“到“算法能效“的进化之路
人工智能·物联网
破无差41 分钟前
人工智能训练师(三级)理论知识复习题-KimiK3的参考答案
人工智能
网络毒刘44 分钟前
AtomGit Actions + AI 评审草稿:PR 描述自动生成与安全敏感词门禁示例
人工智能·安全