YOLO训练结果

YOLO训练结果

背景

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

安全帽检测

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

跌倒检测

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

计划

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