简介基于YOLOv8的陶瓷器鉴定系统面向高校计算机、人工智能、通信工程等专业学生是一套可直接用于毕业设计或课程设计的完整目标检测方案。资源内置精选数据集、预训练权重、可视化交互界面以及逐步部署教程代码均可正常运行部署后即可体验从模型训练、推理测试到结果可视化的完整流程。系统支持生成核心指标曲线、混淆矩阵、F1分数曲线、精确率-召回率曲线、验证集预测结果与标签分布图便于答辩汇报与效果展示。压缩包共97个文件以70个Python源码文件为主另含4个PyTorch模型权重pt、若干配置与说明文档xml/txt以及一段演示视频mp4整体仅24.21MB结构清晰、迁移方便。目前已有29人学习适合需要快速搭建目标检测应用或完成毕设演示的初学者与进阶开发者借鉴使用。1. 陶瓷器鉴定为什么从Faster R-CNN换到YOLOv8答辩现场最怕的一件事是评委指着一堆检测框问“你这个模型为什么选它”。陶瓷器鉴定这个任务本质是五类常见品类的目标检测特征集中在釉色、纹饰和器型轮廓上既要有全局观察又要有局部细节早期我用Faster R-CNN跑过一版单张推理在GTX 1660 Ti上要两百多毫秒小目标比如瓶口开片纹经常漏检调了anchor大小也没有质的改观。换到YOLOv8之后同样的数据集下mAP50提升了大概6个点推理时间掉到原来的三分之一。这篇就围绕这套基于YOLOv8的陶瓷器鉴定系统展开包含完整源码、五类陶瓷器数据集、Tkinter可视化界面和部署教程适合做毕业设计、课程设计或者你手里正缺一个能跑通的深度学习基线项目。它的价值不只是“能跑”而是训练、验证、界面、报告图表这一整条链路都是闭环的。2. 五类陶瓷器数据集落盘与预处理从原始图片到YOLO格式2.1 数据集的目录组织与label格式这套资源里的数据集是标准的YOLO检测格式训练之前我先把它完整梳理了一遍。根目录下一般是images/train、images/val和labels/train、labels/val这样的对位结构每张jpg图片对应一个同名.txt标注文件。标注文件里每一行对应一个目标框格式是class_id x_center y_center width height四个坐标值都是相对图片宽高的归一化结果范围在 0~1 之间。比如下面这行代表类别0的一个目标框中心在图片的(0.52, 0.48)位置宽高分别是图片的0.3和0.350 0.5201 0.4812 0.3012 0.3512拿到别人给的数据集第一步就是写个小脚本检查这些坐标有没有越界、类别id是否连续、标注文件是否有空文件。我一般会跑一段统计代码from pathlib import Path import numpy as np label_dir Path(labels/train) stats {total_boxes: 0, empty_files: 0, out_of_range: 0, class_ids: set()} for f in label_dir.glob(*.txt): lines f.read_text().strip().splitlines() if not lines: stats[empty_files] 1 continue for line in lines: parts list(map(float, line.split())) cls_id int(parts[0]) stats[class_ids].add(cls_id) xc, yc, w, h parts[1:] if xc 0 or yc 0 or xc 1 or yc 1 or w 0 or h 0: stats[out_of_range] 1 stats[total_boxes] 1 print(stats)这段代码把每个标注文件逐行读取统计样本总量、空标注文件和越界框数量同时收集类别ID是否从0连续排列。YOLOv8要求类别ID从0开始连续递增如果数据里有跳号或者标签从1开始训练时会直接报错或者类别错位。这是拿到任何数据集后我必做的一次体检。2.2 类别映射与train/val划分陶瓷器五类在模型里对应的是class_id 0~4训练前需要在data.yaml里把类别名称写死这个文件的顺序就是模型输出的顺序后面可视化界面里的中文标签也是靠它映射还原的path: ./datasets/ceramic train: images/train val: images/val nc: 5 names: 0: qinghua 1: fencai 2: youlihong 3: jinse 4: doucainames的索引顺序必须和标注文件里的class_id完全对应训练跟推理共用同一份yaml否则会出现训练时是青花、推理时标到粉彩头上的错位。原数据集里如果只有训练集没有验证集可以用下面的脚本按比例划分同时保持图片和标注文件的配对关系python - EOF import random from pathlib import Path random.seed(42) src_images Path(all_images) src_labels Path(all_labels) out Path(datasets/ceramic) for split, ratio in [(train, 0.85), (val, 0.15)]: (out / images / split).mkdir(parentsTrue, exist_okTrue) (out / labels / split).mkdir(parentsTrue, exist_okTrue) imgs list(src_images.glob(*.jpg)) random.shuffle(imgs) val_count int(len(imgs) * 0.15) for split, pool in [(train, imgs[val_count:]), (val, imgs[:val_count])]: for img in pool: label src_labels / (img.stem .txt) if not label.exists(): continue img.rename(out / images / split / img.name) label.rename(out / labels / split / label.name) EOF这段脚本的思路是先把全量图片打乱再按85%/15%切分然后移动图片和对应标注文件。seed(42)固定随机种子保证每次划分结果一致这在复现实验和答辩说明“我们是随机划分、没有泄露”时有据可查。注意这里用的是rename跨目录会直接搬文件磁盘空间紧张时比复制省空间。2.3 数据增强参数与陶瓷器场景的隐含坑YOLOv8的数据增强默认是写在augmentations.py里的训练命令可以通过hsv_h、degrees、mosaic这些参数直接覆盖默认值。对于陶瓷器这个场景几个关键参数需要特殊处理参数默认值陶瓷器建议值理由degrees0.05~10陶瓷器照片拍摄角度基本固定小角度旋转模拟手持误差超过15度会产生大量畸形框fliplr0.50.5青花缠枝纹左右翻转不影响类别语义可以用mosaic1.00.8~1.0拼接裁剪能提升小目标能力但会破坏器型完整轮廓建议开到0.8留一点原始样本hsv_h / hsv_s / hsv_v0.015 / 0.7 / 0.4饱和度偏移调小到0.3釉色是鉴定核心特征色相偏移过大会让青花变成近似黑色的错误样本mixup0.00.2陶瓷器背景相对干净mixup带来的背景噪声对特征学习帮助有限可开可不这里最大的坑是degrees。我实际训练时一开始开了30度旋转发现验证集mAP反而掉了因为陶瓷器是三维器形平面旋转后瓶身弧面光影关系错乱釉色特征被破坏。五类样本里青花和粉彩靠的就是色彩分布区分色彩增强参数必须保守。另外陶瓷器数据普遍存在类别不均衡如果五类样本量差异超过3倍建议在训练时给少样本类别乘以一个损失权重YOLOv8里可以用class_weights传入或者在自定义loss里做focal weighting后者在loss.py里直接改减号前的系数就行。3. YOLOv8模型装配与训练参数配置从C2f到Anchor-Free3.1 选YOLOv8而不是YOLOv5的模型结构原因YOLOv8相对YOLOv5最大的改动是C3模块换成了C2f结构backbone提特征时多了一路梯度流分支每层的梯度能更直接地回传到前面的卷积。对陶瓷器这个任务的意义在于纹饰细节集中在器物的局部区域深层特征如果没有浅层的梯度直通细节对比度会弱。C2f的分支设计让网络在训练前期就能更快地关注到这些细粒度纹理。另一个关键改动是检测头从Anchor-Based换成了Anchor-Free配合的解耦头把分类和回归拆成两个独立分支。陶瓷器的形状差异很大梅瓶细长、碗碟矮扁预设anchor很难覆盖这种高宽比跨度Anchor-Free按中心点距离回归天然不依赖预设框这也是它比Faster R-CNN在器型变化大时更稳的原因。原包config里还有rtmdet_m和faster-rcnn的配置文件说明作者做过对比实验这个在毕设论文里完全可以复用来撑起“方案选型对比”那一节。3.2 训练命令与关键参数拆解原包目录下的train_mode.py是训练入口本质是对ultralytics库的训练参数做了一次封装。等效的YOLOv8训练命令是yolo detect train \ modelyolov8n.pt \ dataceramic.yaml \ epochs150 \ imgsz640 \ batch16 \ device0 \ optimizerauto \ lr00.01 \ mosaic0.9 \ patience20 \ projectruns/train \ nameceramic_yolov8n逐项说明modelyolov8n.pt是预训练权重陶瓷器数据量不到几千张时从COCO权重热启动特征提取层已经学会了通用边缘纹理收敛速度远快于从零训练。imgsz640是输入分辨率图片先等比缩放再填充到640x640。batch16在GTX 1660 Ti 6G显存下配合AMP刚好是临界值显存不足就降到8。optimizerauto会自动在SGD和AdamW之间选数据量小时选AdamW收敛更稳。patience20表示连续20个epoch验证集mAP没有提升就提前停止防止样本少时过拟合产生无效训练时间。这个系统里我实际用的还是封装在train_mode.py里的方式它在命令之上额外做了两件事一是训练结束后自动把best.pt复制到model/目录方便后续界面程序引用二是用best.pt跑一遍验证集并输出全部指标图。如果你的显存连batch8都跑不动可以加一行ampTrue打开混合精度1660 Ti对FP16支持正常显存占用能再降三分之一。3.3 训练过程日志与YOLO11对比实验设计训练过程中要盯三个量box_loss、cls_loss和mAP50。正常的曲线是box_loss和cls_loss单调下降后趋于平缓mAP50稳步上升最后在某个值附近震荡。如果cls_loss下降而box_loss不动大概率是框回归分支的学习率被解耦头压制可以单独把box_lr调高。如果mAP曲线来回抖动幅度超过5个点一般是batch太小导致梯度噪声过大。原包目录里还有yolo11n.pt这个文件我推测作者在后续迭代里用YOLO11做过对比实验。这是一个很好的论文素材同一份数据、同样的训练参数只替换modelyolo11n.pt把YOLOv8和YOLO11的mAP与参数量放在同一张表里。YOLO11的backbone在C2f基础上又加重了注意力分支陶瓷器细节纹理的响应会更集中但训练时间也相应变长。如果deadline紧张建议还是以YOLOv8结果为主YOLO11只做加分对比。默认的optimizerauto会在前几个epoch自动warmup不要手动把lr0调到0.1以上陶瓷器数据量小学习率过大很容易在20个epoch内直接发散。4. 可视化界面与推理服务main.py的检测链路与metrics曲线4.1 推理服务封装方式原包里的five_type_det_service.py是核心推理服务它把YOLOv8的推理过程封装成独立的函数界面层不直接接触模型细节。这个思路很值得学界面调用和模型推理解耦后把best.pt换掉或者把后端换成TensorRT都不用改UI代码。简化后的推理逻辑是from ultralytics import YOLO class CeramicDetector: def __init__(self, weightsmodel/best.pt, conf0.25, iou0.45): self.model YOLO(weights) self.conf conf self.iou iou def predict(self, img_path): results self.model.predict( sourceimg_path, confself.conf, iouself.iou, imgsz640, device0, verboseFalse ) boxes results[0].boxes return { boxes: boxes.xyxy.cpu().numpy(), scores: boxes.conf.cpu().numpy(), cls_ids: boxes.cls.cpu().numpy().astype(int) }result self.model.predict(...)内部完成了图像预处理、推理、NMS后处理boxes.xyxy是每个目标框的左上右下角坐标conf是置信度阈值低于0.25的框直接丢掉。iou是NMS的IoU阈值调到0.45对陶瓷器这种存在叠放遮挡的场景更合适调到0.7会把同一个瓶子的多个冗余框合并不干净。verboseFalse是关闭推理过程中的逐帧打印否则视频模式下拉一条终端清单严重影响体验。4.2 Tkinter界面与视频检测链路main.py是基于Tkinter的可视化界面左侧放功能按钮右侧放检测结果图像。核心交互逻辑是三件事点击“选择图片”调用filedialog.askopenfilename拿图片路径传给CeramicDetector.predict推理再把检测框画到图上显示点击“开始检测视频”则走Detection_video.py里的逐帧处理逻辑点击“摄像头”实测则是调用OpenCV的cv2.VideoCapture(0)读取摄像头帧逐帧送进同一个predict函数。三种输入共用同一套推理接口这个设计让新加视频流支持时不需要动任何检测逻辑。界面上的“类别”下拉框是拿data.yaml里的names映射生成的选中某个类别后程序会标注出这一帧里所有该类目标的置信度均值。这部分代码在my_func.py里逻辑就是一个计数器累加每帧的类别置信度然后求平均。置信度低于0.3的框在界面上用浅色标注高于0.7用深色标注用于快速判断模型的置信区间分布。4.3 训练指标图的生成与存放位置这套资源最有答辩价值的是它自动生成的一整套指标图。train_mode.py跑完训练后会在runs/detect/name/目录下生成图表对应文件看图重点损失曲线图results.png训练/验证loss是否同步下降有无发散混淆矩阵confusion_matrix.png哪两类互混严重是否要补数据F1分数曲线F1_curve.png不同置信度阈值下的F1最优区间PR曲线PR_curve.pngAP值近似曲线下面积曲线越靠右上越好验证集预测结果val_batch0_pred.jpg直观检查检测框和类别是否正确标签分布图labels.jpg检查目标大小与位置分布是否畸形换数据集复现的时候只需要保证train_mode.py里的data_config指向新的data.yaml训练完成后这些图会自动覆盖到新的项目目录。如果只跑了训练没有出图可以单独执行验证命令补全yolo detect val \ modelruns/detect/ceramic_yolov8n/weights/best.pt \ dataceramic.yaml \ projectresults_exp \ plotsTrueplotsTrue是输出全部图表的关键开关不加它只返回一行指标文字。答辩PPT里最有用的是混淆矩阵和PR曲线这两张它们分别从“错在哪”和“置信度评估”两个维度解释模型能力比只贴一张loss下降图有说服力得多。5. 从指标到答辩证据链混淆矩阵、PR曲线与损失函数怎么读5.1 混淆矩阵怎么看局部混淆混淆矩阵的对角线是分类正确的比例重点看非对角线的大数字。陶瓷器五类里青花和粉彩互混最常见因为青花瓷在光照偏暖时色调会向粉彩偏移。如果矩阵里这两类互混超过15%直接去采集更多严格打光的样本优先补粉彩弱光环境下的图而不是盲目加大模型。标注误差导致的混淆特征是在对角线两侧对称出现而真实类别界限模糊导致的混淆通常偏向某一侧这个细节评委问起来能讲出东西。5.2 损失函数曲线的四种病态形态看results.png里的损失曲线训练集loss持续下降、验证集loss在第40个epoch后掉头上扬这是过拟合处理办法是加大mixup到0.3或者减少epoch数。训练集loss和验证集loss同步停在同一个高位不动说明模型容量不够把yolov8n.pt换成yolov8s.pt再训。如果loss曲线出现周期性的尖峰然后回落是batch内混入了极端标注样本去检查label文件里是否有异常宽高的框。最后一种训练集loss正常但mAP曲线在50%附近水平震荡重点检查degrees或hsv增强是否把特征破坏得太严重调小增强参数重新训。5.3 部署避坑与快速验证界面程序跑不起来时按顺序排查三个地方。第一main.py和five_type_det_service.py的文件路径里如果包含中文或者空格os.path会产生路径拼接错误原包解压后我建议直接放在E:/ceramic_system这种全英文路径下。第二半精度推理在部分老驱动上会报显存错误把predict参数halfFalse强制关闭。第三类别名字显示错乱时检查data.yaml的names顺序是否和训练时一致Tkinter界面上的下拉框是按这个文件里的顺序渲染的。最后给你一个可以立刻验证模型是否正常的命令用训练出的权重跑一遍验证集看整体mAP是否落在训练日志的合理区间内yolo detect val \ modelmodel/best.pt \ dataceramic.yaml \ batch16 \ device0 \ plotsTrue \ projectfinal_check这条命令跑完花不到两分钟输出的final_check/results.csv里直接查metrics/mAP50(B)和metrics/mAP50-95(B)两列如果和训练结束时的验证精度偏差在1个点以内后续的系数调整和对比实验就都可以放心在这套模型基础上展开。本文还有配套的精品资源点击获取 基于YOLOv8的陶瓷器鉴定系统：数据集训练与可视化部署全流程
简介基于YOLOv8的陶瓷器鉴定系统面向高校计算机、人工智能、通信工程等专业学生是一套可直接用于毕业设计或课程设计的完整目标检测方案。资源内置精选数据集、预训练权重、可视化交互界面以及逐步部署教程代码均可正常运行部署后即可体验从模型训练、推理测试到结果可视化的完整流程。系统支持生成核心指标曲线、混淆矩阵、F1分数曲线、精确率-召回率曲线、验证集预测结果与标签分布图便于答辩汇报与效果展示。压缩包共97个文件以70个Python源码文件为主另含4个PyTorch模型权重pt、若干配置与说明文档xml/txt以及一段演示视频mp4整体仅24.21MB结构清晰、迁移方便。目前已有29人学习适合需要快速搭建目标检测应用或完成毕设演示的初学者与进阶开发者借鉴使用。1. 陶瓷器鉴定为什么从Faster R-CNN换到YOLOv8答辩现场最怕的一件事是评委指着一堆检测框问“你这个模型为什么选它”。陶瓷器鉴定这个任务本质是五类常见品类的目标检测特征集中在釉色、纹饰和器型轮廓上既要有全局观察又要有局部细节早期我用Faster R-CNN跑过一版单张推理在GTX 1660 Ti上要两百多毫秒小目标比如瓶口开片纹经常漏检调了anchor大小也没有质的改观。换到YOLOv8之后同样的数据集下mAP50提升了大概6个点推理时间掉到原来的三分之一。这篇就围绕这套基于YOLOv8的陶瓷器鉴定系统展开包含完整源码、五类陶瓷器数据集、Tkinter可视化界面和部署教程适合做毕业设计、课程设计或者你手里正缺一个能跑通的深度学习基线项目。它的价值不只是“能跑”而是训练、验证、界面、报告图表这一整条链路都是闭环的。2. 五类陶瓷器数据集落盘与预处理从原始图片到YOLO格式2.1 数据集的目录组织与label格式这套资源里的数据集是标准的YOLO检测格式训练之前我先把它完整梳理了一遍。根目录下一般是images/train、images/val和labels/train、labels/val这样的对位结构每张jpg图片对应一个同名.txt标注文件。标注文件里每一行对应一个目标框格式是class_id x_center y_center width height四个坐标值都是相对图片宽高的归一化结果范围在 0~1 之间。比如下面这行代表类别0的一个目标框中心在图片的(0.52, 0.48)位置宽高分别是图片的0.3和0.350 0.5201 0.4812 0.3012 0.3512拿到别人给的数据集第一步就是写个小脚本检查这些坐标有没有越界、类别id是否连续、标注文件是否有空文件。我一般会跑一段统计代码from pathlib import Path import numpy as np label_dir Path(labels/train) stats {total_boxes: 0, empty_files: 0, out_of_range: 0, class_ids: set()} for f in label_dir.glob(*.txt): lines f.read_text().strip().splitlines() if not lines: stats[empty_files] 1 continue for line in lines: parts list(map(float, line.split())) cls_id int(parts[0]) stats[class_ids].add(cls_id) xc, yc, w, h parts[1:] if xc 0 or yc 0 or xc 1 or yc 1 or w 0 or h 0: stats[out_of_range] 1 stats[total_boxes] 1 print(stats)这段代码把每个标注文件逐行读取统计样本总量、空标注文件和越界框数量同时收集类别ID是否从0连续排列。YOLOv8要求类别ID从0开始连续递增如果数据里有跳号或者标签从1开始训练时会直接报错或者类别错位。这是拿到任何数据集后我必做的一次体检。2.2 类别映射与train/val划分陶瓷器五类在模型里对应的是class_id 0~4训练前需要在data.yaml里把类别名称写死这个文件的顺序就是模型输出的顺序后面可视化界面里的中文标签也是靠它映射还原的path: ./datasets/ceramic train: images/train val: images/val nc: 5 names: 0: qinghua 1: fencai 2: youlihong 3: jinse 4: doucainames的索引顺序必须和标注文件里的class_id完全对应训练跟推理共用同一份yaml否则会出现训练时是青花、推理时标到粉彩头上的错位。原数据集里如果只有训练集没有验证集可以用下面的脚本按比例划分同时保持图片和标注文件的配对关系python - EOF import random from pathlib import Path random.seed(42) src_images Path(all_images) src_labels Path(all_labels) out Path(datasets/ceramic) for split, ratio in [(train, 0.85), (val, 0.15)]: (out / images / split).mkdir(parentsTrue, exist_okTrue) (out / labels / split).mkdir(parentsTrue, exist_okTrue) imgs list(src_images.glob(*.jpg)) random.shuffle(imgs) val_count int(len(imgs) * 0.15) for split, pool in [(train, imgs[val_count:]), (val, imgs[:val_count])]: for img in pool: label src_labels / (img.stem .txt) if not label.exists(): continue img.rename(out / images / split / img.name) label.rename(out / labels / split / label.name) EOF这段脚本的思路是先把全量图片打乱再按85%/15%切分然后移动图片和对应标注文件。seed(42)固定随机种子保证每次划分结果一致这在复现实验和答辩说明“我们是随机划分、没有泄露”时有据可查。注意这里用的是rename跨目录会直接搬文件磁盘空间紧张时比复制省空间。2.3 数据增强参数与陶瓷器场景的隐含坑YOLOv8的数据增强默认是写在augmentations.py里的训练命令可以通过hsv_h、degrees、mosaic这些参数直接覆盖默认值。对于陶瓷器这个场景几个关键参数需要特殊处理参数默认值陶瓷器建议值理由degrees0.05~10陶瓷器照片拍摄角度基本固定小角度旋转模拟手持误差超过15度会产生大量畸形框fliplr0.50.5青花缠枝纹左右翻转不影响类别语义可以用mosaic1.00.8~1.0拼接裁剪能提升小目标能力但会破坏器型完整轮廓建议开到0.8留一点原始样本hsv_h / hsv_s / hsv_v0.015 / 0.7 / 0.4饱和度偏移调小到0.3釉色是鉴定核心特征色相偏移过大会让青花变成近似黑色的错误样本mixup0.00.2陶瓷器背景相对干净mixup带来的背景噪声对特征学习帮助有限可开可不这里最大的坑是degrees。我实际训练时一开始开了30度旋转发现验证集mAP反而掉了因为陶瓷器是三维器形平面旋转后瓶身弧面光影关系错乱釉色特征被破坏。五类样本里青花和粉彩靠的就是色彩分布区分色彩增强参数必须保守。另外陶瓷器数据普遍存在类别不均衡如果五类样本量差异超过3倍建议在训练时给少样本类别乘以一个损失权重YOLOv8里可以用class_weights传入或者在自定义loss里做focal weighting后者在loss.py里直接改减号前的系数就行。3. YOLOv8模型装配与训练参数配置从C2f到Anchor-Free3.1 选YOLOv8而不是YOLOv5的模型结构原因YOLOv8相对YOLOv5最大的改动是C3模块换成了C2f结构backbone提特征时多了一路梯度流分支每层的梯度能更直接地回传到前面的卷积。对陶瓷器这个任务的意义在于纹饰细节集中在器物的局部区域深层特征如果没有浅层的梯度直通细节对比度会弱。C2f的分支设计让网络在训练前期就能更快地关注到这些细粒度纹理。另一个关键改动是检测头从Anchor-Based换成了Anchor-Free配合的解耦头把分类和回归拆成两个独立分支。陶瓷器的形状差异很大梅瓶细长、碗碟矮扁预设anchor很难覆盖这种高宽比跨度Anchor-Free按中心点距离回归天然不依赖预设框这也是它比Faster R-CNN在器型变化大时更稳的原因。原包config里还有rtmdet_m和faster-rcnn的配置文件说明作者做过对比实验这个在毕设论文里完全可以复用来撑起“方案选型对比”那一节。3.2 训练命令与关键参数拆解原包目录下的train_mode.py是训练入口本质是对ultralytics库的训练参数做了一次封装。等效的YOLOv8训练命令是yolo detect train \ modelyolov8n.pt \ dataceramic.yaml \ epochs150 \ imgsz640 \ batch16 \ device0 \ optimizerauto \ lr00.01 \ mosaic0.9 \ patience20 \ projectruns/train \ nameceramic_yolov8n逐项说明modelyolov8n.pt是预训练权重陶瓷器数据量不到几千张时从COCO权重热启动特征提取层已经学会了通用边缘纹理收敛速度远快于从零训练。imgsz640是输入分辨率图片先等比缩放再填充到640x640。batch16在GTX 1660 Ti 6G显存下配合AMP刚好是临界值显存不足就降到8。optimizerauto会自动在SGD和AdamW之间选数据量小时选AdamW收敛更稳。patience20表示连续20个epoch验证集mAP没有提升就提前停止防止样本少时过拟合产生无效训练时间。这个系统里我实际用的还是封装在train_mode.py里的方式它在命令之上额外做了两件事一是训练结束后自动把best.pt复制到model/目录方便后续界面程序引用二是用best.pt跑一遍验证集并输出全部指标图。如果你的显存连batch8都跑不动可以加一行ampTrue打开混合精度1660 Ti对FP16支持正常显存占用能再降三分之一。3.3 训练过程日志与YOLO11对比实验设计训练过程中要盯三个量box_loss、cls_loss和mAP50。正常的曲线是box_loss和cls_loss单调下降后趋于平缓mAP50稳步上升最后在某个值附近震荡。如果cls_loss下降而box_loss不动大概率是框回归分支的学习率被解耦头压制可以单独把box_lr调高。如果mAP曲线来回抖动幅度超过5个点一般是batch太小导致梯度噪声过大。原包目录里还有yolo11n.pt这个文件我推测作者在后续迭代里用YOLO11做过对比实验。这是一个很好的论文素材同一份数据、同样的训练参数只替换modelyolo11n.pt把YOLOv8和YOLO11的mAP与参数量放在同一张表里。YOLO11的backbone在C2f基础上又加重了注意力分支陶瓷器细节纹理的响应会更集中但训练时间也相应变长。如果deadline紧张建议还是以YOLOv8结果为主YOLO11只做加分对比。默认的optimizerauto会在前几个epoch自动warmup不要手动把lr0调到0.1以上陶瓷器数据量小学习率过大很容易在20个epoch内直接发散。4. 可视化界面与推理服务main.py的检测链路与metrics曲线4.1 推理服务封装方式原包里的five_type_det_service.py是核心推理服务它把YOLOv8的推理过程封装成独立的函数界面层不直接接触模型细节。这个思路很值得学界面调用和模型推理解耦后把best.pt换掉或者把后端换成TensorRT都不用改UI代码。简化后的推理逻辑是from ultralytics import YOLO class CeramicDetector: def __init__(self, weightsmodel/best.pt, conf0.25, iou0.45): self.model YOLO(weights) self.conf conf self.iou iou def predict(self, img_path): results self.model.predict( sourceimg_path, confself.conf, iouself.iou, imgsz640, device0, verboseFalse ) boxes results[0].boxes return { boxes: boxes.xyxy.cpu().numpy(), scores: boxes.conf.cpu().numpy(), cls_ids: boxes.cls.cpu().numpy().astype(int) }result self.model.predict(...)内部完成了图像预处理、推理、NMS后处理boxes.xyxy是每个目标框的左上右下角坐标conf是置信度阈值低于0.25的框直接丢掉。iou是NMS的IoU阈值调到0.45对陶瓷器这种存在叠放遮挡的场景更合适调到0.7会把同一个瓶子的多个冗余框合并不干净。verboseFalse是关闭推理过程中的逐帧打印否则视频模式下拉一条终端清单严重影响体验。4.2 Tkinter界面与视频检测链路main.py是基于Tkinter的可视化界面左侧放功能按钮右侧放检测结果图像。核心交互逻辑是三件事点击“选择图片”调用filedialog.askopenfilename拿图片路径传给CeramicDetector.predict推理再把检测框画到图上显示点击“开始检测视频”则走Detection_video.py里的逐帧处理逻辑点击“摄像头”实测则是调用OpenCV的cv2.VideoCapture(0)读取摄像头帧逐帧送进同一个predict函数。三种输入共用同一套推理接口这个设计让新加视频流支持时不需要动任何检测逻辑。界面上的“类别”下拉框是拿data.yaml里的names映射生成的选中某个类别后程序会标注出这一帧里所有该类目标的置信度均值。这部分代码在my_func.py里逻辑就是一个计数器累加每帧的类别置信度然后求平均。置信度低于0.3的框在界面上用浅色标注高于0.7用深色标注用于快速判断模型的置信区间分布。4.3 训练指标图的生成与存放位置这套资源最有答辩价值的是它自动生成的一整套指标图。train_mode.py跑完训练后会在runs/detect/name/目录下生成图表对应文件看图重点损失曲线图results.png训练/验证loss是否同步下降有无发散混淆矩阵confusion_matrix.png哪两类互混严重是否要补数据F1分数曲线F1_curve.png不同置信度阈值下的F1最优区间PR曲线PR_curve.pngAP值近似曲线下面积曲线越靠右上越好验证集预测结果val_batch0_pred.jpg直观检查检测框和类别是否正确标签分布图labels.jpg检查目标大小与位置分布是否畸形换数据集复现的时候只需要保证train_mode.py里的data_config指向新的data.yaml训练完成后这些图会自动覆盖到新的项目目录。如果只跑了训练没有出图可以单独执行验证命令补全yolo detect val \ modelruns/detect/ceramic_yolov8n/weights/best.pt \ dataceramic.yaml \ projectresults_exp \ plotsTrueplotsTrue是输出全部图表的关键开关不加它只返回一行指标文字。答辩PPT里最有用的是混淆矩阵和PR曲线这两张它们分别从“错在哪”和“置信度评估”两个维度解释模型能力比只贴一张loss下降图有说服力得多。5. 从指标到答辩证据链混淆矩阵、PR曲线与损失函数怎么读5.1 混淆矩阵怎么看局部混淆混淆矩阵的对角线是分类正确的比例重点看非对角线的大数字。陶瓷器五类里青花和粉彩互混最常见因为青花瓷在光照偏暖时色调会向粉彩偏移。如果矩阵里这两类互混超过15%直接去采集更多严格打光的样本优先补粉彩弱光环境下的图而不是盲目加大模型。标注误差导致的混淆特征是在对角线两侧对称出现而真实类别界限模糊导致的混淆通常偏向某一侧这个细节评委问起来能讲出东西。5.2 损失函数曲线的四种病态形态看results.png里的损失曲线训练集loss持续下降、验证集loss在第40个epoch后掉头上扬这是过拟合处理办法是加大mixup到0.3或者减少epoch数。训练集loss和验证集loss同步停在同一个高位不动说明模型容量不够把yolov8n.pt换成yolov8s.pt再训。如果loss曲线出现周期性的尖峰然后回落是batch内混入了极端标注样本去检查label文件里是否有异常宽高的框。最后一种训练集loss正常但mAP曲线在50%附近水平震荡重点检查degrees或hsv增强是否把特征破坏得太严重调小增强参数重新训。5.3 部署避坑与快速验证界面程序跑不起来时按顺序排查三个地方。第一main.py和five_type_det_service.py的文件路径里如果包含中文或者空格os.path会产生路径拼接错误原包解压后我建议直接放在E:/ceramic_system这种全英文路径下。第二半精度推理在部分老驱动上会报显存错误把predict参数halfFalse强制关闭。第三类别名字显示错乱时检查data.yaml的names顺序是否和训练时一致Tkinter界面上的下拉框是按这个文件里的顺序渲染的。最后给你一个可以立刻验证模型是否正常的命令用训练出的权重跑一遍验证集看整体mAP是否落在训练日志的合理区间内yolo detect val \ modelmodel/best.pt \ dataceramic.yaml \ batch16 \ device0 \ plotsTrue \ projectfinal_check这条命令跑完花不到两分钟输出的final_check/results.csv里直接查metrics/mAP50(B)和metrics/mAP50-95(B)两列如果和训练结束时的验证精度偏差在1个点以内后续的系数调整和对比实验就都可以放心在这套模型基础上展开。本文还有配套的精品资源点击获取