YOLO训练结果

YOLO训练结果

背景

之前说,弄好服务器了,然后就准备数据集训练自己的模型的,今天记录下结果。

安全帽检测

安全帽用了数据集将近3000张左右吧,然后训练了两天多,loss最低的时候大概是0.9几。不知道谁kill了我的训练,没有训练到五万多步,但是训练出来test的效果还算可以。
在这里插入图片描述

跌倒检测

Fall detection大概用了2700张数据集吧,数据集的质量不如安全帽的质量高,但是loss在三万多迭代的时候就已经到了0.05左右了,还有12小时训练结束,根据三万多的模型,马上做下test看下效果,我怕不是很好。
在这里插入图片描述

计划

之后要数据可视化一下,然后做下视频流的检测。

### YOLO 训练结果与 get_map 图像差异分析 YOLO训练结果与`get_map`生成的图像存在较大差异可能源于多个方面。这些因素包括但不限于数据集预处理方式的不同、模型配置参数设置不当以及评估指标计算方法的区别。 #### 数据集预处理的影响 在YOLO目标检测框架中,输入到网络的数据通常会经过一系列标准化操作,比如调整大小(resize)、归一化(normalization),甚至随机增强(augmentation)[^1]。如果用于测试或验证阶段通过`get_map`函数获取的地图所基于的原始图像是未经上述相同预处理流程处理过的,则两者之间必然会出现视觉上的不一致现象。 #### 模型预测框与真实标签匹配机制 另一个可能导致这种差距的原因在于YOLO算法内部对于边界框(bounding box)位置和尺寸的具体定义及其与地面实况(ground truth)之间的映射关系上。当采用不同的IoU阈值或者其他相似度量标准来决定哪些预测应该被视作正样本,最终得到的结果也会有所区别[^2]。 #### 解决方案建议 为了缩小YOLO训练输出同由`get_map`绘制出来的可视化效果间的鸿沟: - **统一数据准备过程**:确保所有参与对比实验中的图片都经历了完全相同的前处理步骤; - **校准评价体系**:仔细审查并同步两个不同环节下使用的各类超参设定(如置信度得分阈限confidence score threshold),使得它们尽可能保持一致; - **调试模式下的辅助工具运用**:利用诸如TensorBoard之类的监控平台实观察训练动态变化趋势,并借助混淆矩阵(confusion matrix)等统计图表深入剖析误差分布特征。 ```python import matplotlib.pyplot as plt from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay def plot_confusion_matrix(y_true, y_pred, classes): cm = confusion_matrix(y_true, y_pred) disp = ConfusionMatrixDisplay(cm, display_labels=classes) fig, ax = plt.subplots(figsize=(10, 10)) disp.plot(ax=ax) plt.show() ```
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值