简介这份资源是面向目标检测初学者与计算机视觉方向开发者的篮球运动员检测YOLO格式数据集可直接用于PyTorch框架下的模型训练与算法验证。数据采集自篮球比赛视频与图片覆盖不同场景、角度和光照条件并经过人工标注与格式转换标注类别为篮球运动员适合作为目标检测课程作业、毕业设计或算法对比实验的实战素材。压缩包共2000个文件包含1484个txt标注文件、514张jpg图像以及2个yaml配置文件整体约122.82MB其中txt对应YOLO格式的边界框与类别信息jpg为原始训练图像yaml用于定义数据集路径与类别名称。目前已有1037人学习下载读者可借此快速搭建训练、验证与测试流程省去从零采集与标注的成本并在此基础上调整网络结构或超参数观察模型在真实篮球场景中的检测表现。1. 篮球运动员检测数据集从拿到压缩包到跑通第一帧检测篮球运动员检测这件事难的不是模型结构而是数据。场上十个人挤在三秒区球衣颜色接近、肢体互相遮挡、跑动带来运动模糊再加上替补席和观众席的干扰通用 COCO 预训练权重直接推理经常把裁判认成球员、把篮球架认成人。Basketball-Players-yolov5pytorch.zip 这类数据集的价值就在这里它已经把标注整理成 YOLO 训练所需的 txt 格式目录结构对齐 yolov5 的 images/labels 约定省掉了从 VOC、COCO 转换的那一步。这篇文章面向的是想用 YOLO 做篮球场景检测的从业者和学生从解压校验、目录规整、data.yaml 配置一路讲到训练参数、显存边界和推理验证。读完你应该能自己判断这份数据集够不够用、要不要补标、以及怎么在 PyTorch 环境下把它跑起来。2. 先看清数据集YOLO 格式到底长什么样2.1 目录结构与标注文件的真实形态拿到 Basketball-Players-yolov5pytorch.zip 之后第一件事不是急着训练而是解压后把目录结构看清楚。YOLO 格式的数据集通常长这样unzip Basketball-Players-yolov5pytorch.zip -d basketball_dataset cd basketball_dataset find . -maxdepth 3 -type d | sort常见输出会是 train、valid、test 三个子集每个子集下面各有 images 和 labels 两个文件夹。images 里是 jpg 或 pnglabels 里是同名的 txt。这里有个血泪经验很多打包好的数据集会把 images 和 labels 放在同一层或者 labels 里混进了 classes.txt、notes.txt 这类非标注文件直接喂给 yolov5 会在扫描阶段报错或者静默跳过。先做一次文件配对检查import os from pathlib import Path root Path(basketball_dataset) for split in [train, valid, test]: img_dir root / split / images lbl_dir root / split / labels if not img_dir.exists(): print(f[跳过] {split} 不存在) continue imgs {p.stem for p in img_dir.glob(*) if p.suffix.lower() in (.jpg, .jpeg, .png)} lbls {p.stem for p in lbl_dir.glob(*.txt)} only_img imgs - lbls only_lbl lbls - imgs print(f{split}: 图片 {len(imgs)} 张, 标注 {len(lbls)} 个, 缺标注 {len(only_img)}, 缺图片 {len(only_lbl)})这段脚本做的是集合差运算。imgs 和 lbls 分别收集图片和标注的主文件名相减就能找出孤儿文件。缺标注的图片如果直接参与训练yolov5 会把它当成纯背景图负样本少量可以接受大量说明数据打包有问题。缺图片的标注则是废文件建议直接删掉否则 dataloader 扫描时会反复告警。2.2 标注内容校验类别 id、归一化坐标与越界YOLO 的标注格式是每行class_id x_center y_center width height后四个都是相对图像宽高的归一化值范围 0 到 1。篮球数据集里类别通常只有一到两类比如 player 和 ball或者干脆只有 player。校验脚本要盯三件事类别 id 是否越界、坐标是否超出 [0,1]、宽高是否为 0。import numpy as np def check_labels(lbl_dir, num_classes1): bad_cls, bad_coord, empty 0, 0, 0 for txt in Path(lbl_dir).glob(*.txt): lines txt.read_text().strip().splitlines() if not lines: empty 1 continue for line in lines: parts line.split() if len(parts) ! 5: bad_coord 1 continue cid int(float(parts[0])) vals np.array([float(x) for x in parts[1:]]) if cid 0 or cid num_classes: bad_cls 1 if (vals 0).any() or (vals 1).any() or (vals[2:] 0).any(): bad_coord 1 print(f类别越界 {bad_cls}, 坐标异常 {bad_coord}, 空标注 {empty}) check_labels(basketball_dataset/train/labels, num_classes1)参数说明num_classes 要和你 data.yaml 里写的 nc 一致写错了会把正常类别判成越界。坐标异常里最常见的是 x_center 略微超过 1通常是标注工具在图像边缘框选时产生的浮点误差可以裁剪到 [0,1] 而不是直接丢弃。空标注文件是合法的代表这张图没有目标属于负样本但比例别超过总样本的 10%否则模型会偏向预测背景。2.3 类别分布统计别让某一类把模型带偏篮球场景里如果同时标了 player 和 ballball 的数量往往只有 player 的几十分之一这种长尾分布会让模型几乎不预测球。统计一下每类的框数from collections import Counter counter Counter() for txt in Path(basketball_dataset/train/labels).glob(*.txt): for line in txt.read_text().strip().splitlines(): if line: counter[int(float(line.split()[0]))] 1 print(counter)如果发现 ball 类样本极少常见做法是要么在 data.yaml 里只保留 player 一类把问题简化成单类检测要么对含 ball 的图片做过采样用 yolov5 的--weights配合自定义 sampler或者干脆复制这些图片到训练集里提高出现频率。我一般会先跑单类版本确认 pipeline 通了再决定要不要加球类。3. 环境与训练把 yolov5 跑起来的最小路径3.1 PyTorch 环境搭建与版本匹配yolov5 对 PyTorch 版本比较敏感太新或太旧都可能踩坑。稳妥的组合是 Python 3.8 到 3.10PyTorch 1.13 到 2.1CUDA 11.7 或 11.8。用 conda 建环境conda create -n yolo_basket python3.9 -y conda activate yolo_basket pip install torch2.0.1 torchvision0.15.2 --index-url https://download.pytorch.org/whl/cu118 git clone https://github.com/ultralytics/yolov5 cd yolov5 pip install -r requirements.txt这里有个高频翻车点conda activate报「无法将 conda 项识别」多半是 conda 没初始化 shell跑一次conda init bashWindows 下是conda init powershell重开终端即可。另一个坑是 pip 装的 torch 和 conda 环境里的 numpy 版本冲突表现为ImportError: numpy.core.multiarray failed to import解决方法是pip install numpy2。装完用一行命令验证python -c import torch; print(torch.__version__, torch.cuda.is_available())输出 True 才说明 GPU 可用False 的话检查显卡驱动和 CUDA 版本是否匹配别硬着头皮用 CPU 训篮球数据集几百张图 CPU 训一轮要几十分钟。3.2 data.yaml 配置与路径陷阱yolov5 靠 data.yaml 找数据。在数据集根目录建一个path: /abs/path/to/basketball_dataset train: train/images val: valid/images test: test/images nc: 1 names: [player]参数说明path 必须是绝对路径相对路径在不同工作目录下启动训练会找不到文件这是新手最常踩的坑。train/val/test 写相对于 path 的子路径。nc 是类别数names 的顺序必须和标注里的 class_id 对应写反了模型会把球员标成球。如果数据集只有 train 和 valid把 test 那行删掉别留空值。3.3 训练命令与关键超参数最小训练命令python train.py \ --img 640 \ --batch 16 \ --epochs 100 \ --data /abs/path/to/basketball_dataset/data.yaml \ --weights yolov5s.pt \ --project runs/train \ --name basketball_v1参数逐个说--img 640是输入分辨率篮球场景里球员占画面比例不小640 够用显存紧张可以降到 416但小目标远处的球会丢。--batch 16是批大小8G 显存跑 yolov5s 加 640 分辨率大概能到 16爆显存就减半。--weights yolov5s.pt用官方预训练权重做迁移学习比从头训收敛快得多数据量小于两千张时几乎必须这么做。--epochs 100对几百到几千张图通常够看验证集 mAP 曲线连续 20 轮不涨就可以停。训练过程中重点看三个指标box_loss 持续下降说明定位在学obj_loss 下降说明目标置信度在收敛mAP0.5 是最终效果。如果 box_loss 降但 mAP 不涨多半是过拟合加--dropout 0.2或者做数据增强。yolov5 默认开了 mosaic、HSV 抖动、随机翻转篮球场景里左右翻转是安全的上下翻转会让球员倒立建议在 hyp 文件里把flipud设为 0。4. 避坑与排查训练篮球数据集时最容易翻车的五件事4.1 现象训练一开始就报「No labels found」原因data.yaml 里的路径写错或者 labels 目录名不是 yolov5 期望的。yolov5 会把 images 路径里的images替换成labels去找标注如果你的目录叫label或者annotations它就找不到。解决要么改目录名对齐约定要么在 data.yaml 里显式写 labels 路径。跑一次python utils/general.py里的检查逻辑或者手动确认train/images和train/labels同级存在。4.2 现象mAP 一直是 0loss 却在降原因类别 id 和 names 对不上。比如标注里 player 是 0但 data.yaml 里 names 写成了[ball, player]模型学的是 ball 对应 0验证时按 player 去匹配就全错。解决统计标注里出现的所有 class_id确保 data.yaml 的 names 列表长度等于 nc且顺序和 id 一致。单类数据集直接写names: [player]。4.3 现象显存溢出报 CUDA out of memory原因batch 太大、img 太大或者 dataloader 的 workers 太多导致内存碎片。篮球数据集图片如果分辨率很高比如 1920x1080yolov5 会先缩放到 img 尺寸但缩放前的解码也占内存。解决先把 batch 降到 8 或 4再把--workers从默认 8 降到 2。还不行就降 img 到 416。另外确认没有其他进程占着 GPUnvidia-smi看一眼。4.4 现象验证集效果好实际推理一塌糊涂原因训练集和验证集来自同一批视频的相邻帧画面高度相似验证集 mAP 虚高。真实场景换了个球馆、换了球衣颜色模型就崩。解决划分数据集时按视频或按场次切分别按帧随机切。如果数据集已经切好自己重新按来源分组。另外推理时把 conf 阈值从默认 0.25 调到 0.4 到 0.5能过滤掉不少误检。4.5 现象检测框抖动严重同一球员帧间跳来跳去原因单帧检测没有时序约束篮球运动员快速跑动时相邻帧的框位置差异大。这是检测模型的固有局限不是训练没训好。解决后处理加跟踪常见做法是接 ByteTrack 或 BoT-SORT用卡尔曼滤波平滑轨迹。yolov5 生态里有现成的 tracking 脚本把检测结果喂进去就能拿到稳定的 id。如果只是做统计人数跟踪还能避免重复计数。5. 进阶技巧用验证脚本量化模型到底行不行训练完别只看终端打印的 mAP自己写一个验证脚本把预测结果和真值对齐看才能发现模型到底错在哪。下面这段代码加载训练好的权重对验证集逐图推理统计漏检和误检import torch from pathlib import Path from PIL import Image model torch.hub.load(ultralytics/yolov5, custom, pathruns/train/basketball_v1/weights/best.pt) model.conf 0.4 model.iou 0.45 val_imgs list(Path(basketball_dataset/valid/images).glob(*.jpg)) miss, false_pos, total_gt 0, 0, 0 for img_path in val_imgs[:200]: results model(Image.open(img_path)) preds results.xyxy[0].cpu().numpy() lbl_path Path(str(img_path).replace(images, labels)).with_suffix(.txt) gt [l for l in lbl_path.read_text().strip().splitlines() if l] if lbl_path.exists() else [] total_gt len(gt) if len(gt) 0 and len(preds) 0: miss 1 if len(gt) 0 and len(preds) 0: false_pos 1 print(f漏检图 {miss}, 误检图 {false_pos}, 真值框总数 {total_gt})逻辑说明model.conf 是置信度阈值调高减少误检但增加漏检篮球场景建议 0.4 起步。model.iou 是 NMS 的 IoU 阈值球员密集时调低到 0.4 能减少框合并。脚本只统计了「整图漏检」和「整图误检」两种粗粒度错误更细的可以算每个框的 IoU 匹配。跑完如果漏检图占比超过 15%说明模型对某些姿态或遮挡场景学得不够需要针对性补标这类样本再微调。几个实操习惯第一每次改完 data.yaml 或 hyp 参数先跑--epochs 1确认 pipeline 通别一上来就 100 轮浪费一晚上发现路径错了。第二权重文件按数据集版本_模型_日期命名不然一周后你分不清哪个 best.pt 是哪个实验的。第三推理阈值别照搬默认值拿验证集画一条 conf 从 0.1 到 0.7 的 precision-recall 曲线选 F1 最高的点。这套流程我跑过好几个检测项目篮球数据集的坑基本集中在路径、类别对齐和过拟合三处把这三处盯住剩下的就是调参耐心活。希望帮到你。本文还有配套的精品资源点击获取
简介这份资源是面向目标检测初学者与计算机视觉方向开发者的篮球运动员检测YOLO格式数据集可直接用于PyTorch框架下的模型训练与算法验证。数据采集自篮球比赛视频与图片覆盖不同场景、角度和光照条件并经过人工标注与格式转换标注类别为篮球运动员适合作为目标检测课程作业、毕业设计或算法对比实验的实战素材。压缩包共2000个文件包含1484个txt标注文件、514张jpg图像以及2个yaml配置文件整体约122.82MB其中txt对应YOLO格式的边界框与类别信息jpg为原始训练图像yaml用于定义数据集路径与类别名称。目前已有1037人学习下载读者可借此快速搭建训练、验证与测试流程省去从零采集与标注的成本并在此基础上调整网络结构或超参数观察模型在真实篮球场景中的检测表现。1. 篮球运动员检测数据集从拿到压缩包到跑通第一帧检测篮球运动员检测这件事难的不是模型结构而是数据。场上十个人挤在三秒区球衣颜色接近、肢体互相遮挡、跑动带来运动模糊再加上替补席和观众席的干扰通用 COCO 预训练权重直接推理经常把裁判认成球员、把篮球架认成人。Basketball-Players-yolov5pytorch.zip 这类数据集的价值就在这里它已经把标注整理成 YOLO 训练所需的 txt 格式目录结构对齐 yolov5 的 images/labels 约定省掉了从 VOC、COCO 转换的那一步。这篇文章面向的是想用 YOLO 做篮球场景检测的从业者和学生从解压校验、目录规整、data.yaml 配置一路讲到训练参数、显存边界和推理验证。读完你应该能自己判断这份数据集够不够用、要不要补标、以及怎么在 PyTorch 环境下把它跑起来。2. 先看清数据集YOLO 格式到底长什么样2.1 目录结构与标注文件的真实形态拿到 Basketball-Players-yolov5pytorch.zip 之后第一件事不是急着训练而是解压后把目录结构看清楚。YOLO 格式的数据集通常长这样unzip Basketball-Players-yolov5pytorch.zip -d basketball_dataset cd basketball_dataset find . -maxdepth 3 -type d | sort常见输出会是 train、valid、test 三个子集每个子集下面各有 images 和 labels 两个文件夹。images 里是 jpg 或 pnglabels 里是同名的 txt。这里有个血泪经验很多打包好的数据集会把 images 和 labels 放在同一层或者 labels 里混进了 classes.txt、notes.txt 这类非标注文件直接喂给 yolov5 会在扫描阶段报错或者静默跳过。先做一次文件配对检查import os from pathlib import Path root Path(basketball_dataset) for split in [train, valid, test]: img_dir root / split / images lbl_dir root / split / labels if not img_dir.exists(): print(f[跳过] {split} 不存在) continue imgs {p.stem for p in img_dir.glob(*) if p.suffix.lower() in (.jpg, .jpeg, .png)} lbls {p.stem for p in lbl_dir.glob(*.txt)} only_img imgs - lbls only_lbl lbls - imgs print(f{split}: 图片 {len(imgs)} 张, 标注 {len(lbls)} 个, 缺标注 {len(only_img)}, 缺图片 {len(only_lbl)})这段脚本做的是集合差运算。imgs 和 lbls 分别收集图片和标注的主文件名相减就能找出孤儿文件。缺标注的图片如果直接参与训练yolov5 会把它当成纯背景图负样本少量可以接受大量说明数据打包有问题。缺图片的标注则是废文件建议直接删掉否则 dataloader 扫描时会反复告警。2.2 标注内容校验类别 id、归一化坐标与越界YOLO 的标注格式是每行class_id x_center y_center width height后四个都是相对图像宽高的归一化值范围 0 到 1。篮球数据集里类别通常只有一到两类比如 player 和 ball或者干脆只有 player。校验脚本要盯三件事类别 id 是否越界、坐标是否超出 [0,1]、宽高是否为 0。import numpy as np def check_labels(lbl_dir, num_classes1): bad_cls, bad_coord, empty 0, 0, 0 for txt in Path(lbl_dir).glob(*.txt): lines txt.read_text().strip().splitlines() if not lines: empty 1 continue for line in lines: parts line.split() if len(parts) ! 5: bad_coord 1 continue cid int(float(parts[0])) vals np.array([float(x) for x in parts[1:]]) if cid 0 or cid num_classes: bad_cls 1 if (vals 0).any() or (vals 1).any() or (vals[2:] 0).any(): bad_coord 1 print(f类别越界 {bad_cls}, 坐标异常 {bad_coord}, 空标注 {empty}) check_labels(basketball_dataset/train/labels, num_classes1)参数说明num_classes 要和你 data.yaml 里写的 nc 一致写错了会把正常类别判成越界。坐标异常里最常见的是 x_center 略微超过 1通常是标注工具在图像边缘框选时产生的浮点误差可以裁剪到 [0,1] 而不是直接丢弃。空标注文件是合法的代表这张图没有目标属于负样本但比例别超过总样本的 10%否则模型会偏向预测背景。2.3 类别分布统计别让某一类把模型带偏篮球场景里如果同时标了 player 和 ballball 的数量往往只有 player 的几十分之一这种长尾分布会让模型几乎不预测球。统计一下每类的框数from collections import Counter counter Counter() for txt in Path(basketball_dataset/train/labels).glob(*.txt): for line in txt.read_text().strip().splitlines(): if line: counter[int(float(line.split()[0]))] 1 print(counter)如果发现 ball 类样本极少常见做法是要么在 data.yaml 里只保留 player 一类把问题简化成单类检测要么对含 ball 的图片做过采样用 yolov5 的--weights配合自定义 sampler或者干脆复制这些图片到训练集里提高出现频率。我一般会先跑单类版本确认 pipeline 通了再决定要不要加球类。3. 环境与训练把 yolov5 跑起来的最小路径3.1 PyTorch 环境搭建与版本匹配yolov5 对 PyTorch 版本比较敏感太新或太旧都可能踩坑。稳妥的组合是 Python 3.8 到 3.10PyTorch 1.13 到 2.1CUDA 11.7 或 11.8。用 conda 建环境conda create -n yolo_basket python3.9 -y conda activate yolo_basket pip install torch2.0.1 torchvision0.15.2 --index-url https://download.pytorch.org/whl/cu118 git clone https://github.com/ultralytics/yolov5 cd yolov5 pip install -r requirements.txt这里有个高频翻车点conda activate报「无法将 conda 项识别」多半是 conda 没初始化 shell跑一次conda init bashWindows 下是conda init powershell重开终端即可。另一个坑是 pip 装的 torch 和 conda 环境里的 numpy 版本冲突表现为ImportError: numpy.core.multiarray failed to import解决方法是pip install numpy2。装完用一行命令验证python -c import torch; print(torch.__version__, torch.cuda.is_available())输出 True 才说明 GPU 可用False 的话检查显卡驱动和 CUDA 版本是否匹配别硬着头皮用 CPU 训篮球数据集几百张图 CPU 训一轮要几十分钟。3.2 data.yaml 配置与路径陷阱yolov5 靠 data.yaml 找数据。在数据集根目录建一个path: /abs/path/to/basketball_dataset train: train/images val: valid/images test: test/images nc: 1 names: [player]参数说明path 必须是绝对路径相对路径在不同工作目录下启动训练会找不到文件这是新手最常踩的坑。train/val/test 写相对于 path 的子路径。nc 是类别数names 的顺序必须和标注里的 class_id 对应写反了模型会把球员标成球。如果数据集只有 train 和 valid把 test 那行删掉别留空值。3.3 训练命令与关键超参数最小训练命令python train.py \ --img 640 \ --batch 16 \ --epochs 100 \ --data /abs/path/to/basketball_dataset/data.yaml \ --weights yolov5s.pt \ --project runs/train \ --name basketball_v1参数逐个说--img 640是输入分辨率篮球场景里球员占画面比例不小640 够用显存紧张可以降到 416但小目标远处的球会丢。--batch 16是批大小8G 显存跑 yolov5s 加 640 分辨率大概能到 16爆显存就减半。--weights yolov5s.pt用官方预训练权重做迁移学习比从头训收敛快得多数据量小于两千张时几乎必须这么做。--epochs 100对几百到几千张图通常够看验证集 mAP 曲线连续 20 轮不涨就可以停。训练过程中重点看三个指标box_loss 持续下降说明定位在学obj_loss 下降说明目标置信度在收敛mAP0.5 是最终效果。如果 box_loss 降但 mAP 不涨多半是过拟合加--dropout 0.2或者做数据增强。yolov5 默认开了 mosaic、HSV 抖动、随机翻转篮球场景里左右翻转是安全的上下翻转会让球员倒立建议在 hyp 文件里把flipud设为 0。4. 避坑与排查训练篮球数据集时最容易翻车的五件事4.1 现象训练一开始就报「No labels found」原因data.yaml 里的路径写错或者 labels 目录名不是 yolov5 期望的。yolov5 会把 images 路径里的images替换成labels去找标注如果你的目录叫label或者annotations它就找不到。解决要么改目录名对齐约定要么在 data.yaml 里显式写 labels 路径。跑一次python utils/general.py里的检查逻辑或者手动确认train/images和train/labels同级存在。4.2 现象mAP 一直是 0loss 却在降原因类别 id 和 names 对不上。比如标注里 player 是 0但 data.yaml 里 names 写成了[ball, player]模型学的是 ball 对应 0验证时按 player 去匹配就全错。解决统计标注里出现的所有 class_id确保 data.yaml 的 names 列表长度等于 nc且顺序和 id 一致。单类数据集直接写names: [player]。4.3 现象显存溢出报 CUDA out of memory原因batch 太大、img 太大或者 dataloader 的 workers 太多导致内存碎片。篮球数据集图片如果分辨率很高比如 1920x1080yolov5 会先缩放到 img 尺寸但缩放前的解码也占内存。解决先把 batch 降到 8 或 4再把--workers从默认 8 降到 2。还不行就降 img 到 416。另外确认没有其他进程占着 GPUnvidia-smi看一眼。4.4 现象验证集效果好实际推理一塌糊涂原因训练集和验证集来自同一批视频的相邻帧画面高度相似验证集 mAP 虚高。真实场景换了个球馆、换了球衣颜色模型就崩。解决划分数据集时按视频或按场次切分别按帧随机切。如果数据集已经切好自己重新按来源分组。另外推理时把 conf 阈值从默认 0.25 调到 0.4 到 0.5能过滤掉不少误检。4.5 现象检测框抖动严重同一球员帧间跳来跳去原因单帧检测没有时序约束篮球运动员快速跑动时相邻帧的框位置差异大。这是检测模型的固有局限不是训练没训好。解决后处理加跟踪常见做法是接 ByteTrack 或 BoT-SORT用卡尔曼滤波平滑轨迹。yolov5 生态里有现成的 tracking 脚本把检测结果喂进去就能拿到稳定的 id。如果只是做统计人数跟踪还能避免重复计数。5. 进阶技巧用验证脚本量化模型到底行不行训练完别只看终端打印的 mAP自己写一个验证脚本把预测结果和真值对齐看才能发现模型到底错在哪。下面这段代码加载训练好的权重对验证集逐图推理统计漏检和误检import torch from pathlib import Path from PIL import Image model torch.hub.load(ultralytics/yolov5, custom, pathruns/train/basketball_v1/weights/best.pt) model.conf 0.4 model.iou 0.45 val_imgs list(Path(basketball_dataset/valid/images).glob(*.jpg)) miss, false_pos, total_gt 0, 0, 0 for img_path in val_imgs[:200]: results model(Image.open(img_path)) preds results.xyxy[0].cpu().numpy() lbl_path Path(str(img_path).replace(images, labels)).with_suffix(.txt) gt [l for l in lbl_path.read_text().strip().splitlines() if l] if lbl_path.exists() else [] total_gt len(gt) if len(gt) 0 and len(preds) 0: miss 1 if len(gt) 0 and len(preds) 0: false_pos 1 print(f漏检图 {miss}, 误检图 {false_pos}, 真值框总数 {total_gt})逻辑说明model.conf 是置信度阈值调高减少误检但增加漏检篮球场景建议 0.4 起步。model.iou 是 NMS 的 IoU 阈值球员密集时调低到 0.4 能减少框合并。脚本只统计了「整图漏检」和「整图误检」两种粗粒度错误更细的可以算每个框的 IoU 匹配。跑完如果漏检图占比超过 15%说明模型对某些姿态或遮挡场景学得不够需要针对性补标这类样本再微调。几个实操习惯第一每次改完 data.yaml 或 hyp 参数先跑--epochs 1确认 pipeline 通别一上来就 100 轮浪费一晚上发现路径错了。第二权重文件按数据集版本_模型_日期命名不然一周后你分不清哪个 best.pt 是哪个实验的。第三推理阈值别照搬默认值拿验证集画一条 conf 从 0.1 到 0.7 的 precision-recall 曲线选 F1 最高的点。这套流程我跑过好几个检测项目篮球数据集的坑基本集中在路径、类别对齐和过拟合三处把这三处盯住剩下的就是调参耐心活。希望帮到你。本文还有配套的精品资源点击获取
