YOLOv7 自定义目标检测训练实战：数据准备、训练与推理展示

YOLOv7 自定义目标检测训练实战数据准备、训练与推理展示这篇教程根据我复现 YOLOv7 自定义目标检测流程时整理重点演示依赖安装、数据集准备、模型训练、评估和测试集推理。本文整理自我的学习和项目复现过程尽量按实操顺序保留 notebook 的关键步骤同时把数据集获取方式调整为适合中文教程发布的写法。本文会重点跑通以下流程安装 YOLOv7 环境从数据集后台获取检测数据下载 COCO 预训练权重启动自定义训练评估并展示测试集预测如果你正在系统学习目标检测、实例分割、OCR、多目标跟踪或视觉大模型建议收藏本文配套 notebook、示例图片和运行环境说明后续会继续整理。如果环境配置卡住可以在评论区说明具体报错。 文章目录YOLOv7 自定义目标检测训练实战数据准备、训练与推理展示⚙️ 安装依赖 从数据集后台获取检测数据️ 开始训练 评估模型 部署说明 小结 同系列教程汇总⚙️ 安装依赖下面按 notebook 原流程继续执行。!nvidia-smi# 下载 YOLOv7 仓库并安装依赖# !git clone https://github.com/WongKinYiu/yolov7# %cd yolov7# !pip install -r requirements.txt# current version of YOLOv7 is not compatible with pytorch1.12.1 and numpy1.20.1# until the appropriate changes get made to the main repository, we will be using a fork containing the patched code# you can track the progress here: https://github.com//notebooks/issues/27!git clone https://github.com/SkalskiP/yolov7.git%cd yolov7 !git checkout fix/problems_associated_with_the_latest_versions_of_pytorch_and_numpy !pip install-r requirements.txt 从数据集后台获取检测数据下面按 notebook 原流程继续执行。fromtypesimportSimpleNamespace# 从数据集后台下载 YOLO 检测 格式数据集后修改 DATASET_DIR 指向解压目录。DATASET_DIR/content/dataset# 修改为数据集后台导出的数据集目录datasetSimpleNamespace(locationDATASET_DIR,version1,namecustom-dataset)️ 开始训练下面按 notebook 原流程继续执行。# download COCO starting checkpoint%cd/content/yolov7 !wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7_training.pt# 运行此单元开始训练%cd/content/yolov7 !python train.py--batch16--epochs55--data{dataset.location}/data.yaml--weightsyolov7_training.pt--device0 评估模型下面按 notebook 原流程继续执行。# 运行评估!python detect.py--weights runs/train/exp/weights/best.pt--conf0.1--source{dataset.location}/test/images#display inference on ALL test imagesimportglobfromIPython.displayimportImage,display i0limit10000# max images to printforimageNameinglob.glob(/content/yolov7/runs/detect/exp/*.jpg):#assuming JPGifilimit:display(Image(filenameimageName))print(\n)ii1 部署说明Notebook 原流程包含在线部署。这里保留本地模型路径方便接入自己的部署流程。# 可选将训练得到的 YOLOv7 权重接入自己的部署流程。MODEL_DIR/content/yolov7/runs/trainMODEL_DIR 小结这篇教程完整整理了YOLOv7 自定义目标检测训练的核心复现流程。实际操作时建议先确认 GPU、依赖版本、数据集路径和模型权重路径再逐段运行 notebook。后续我会继续按源项目顺序整理同系列中的目标检测、实例分割、OCR、多目标跟踪和视觉大模型教程。 同系列教程汇总Google Gemini 3.5 Flash 零样本目标检测教程从提示词到可视化结果GLM-OCR 文档识别实战教程从验证码、公式到车牌 OCRRF-DETR ByteTrack 多目标跟踪实战教程从命令行到 Python 视频轨迹可视化SAM 3 图像分割实战教程文本、框和点提示的多种分割方式SAM 3 视频分割实战教程用文本提示分割并跟踪视频中的目标YOLOv7 自定义目标检测训练实战数据准备、训练与推理展示-本文