YOLO训练结果

YOLO训练结果

背景

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

安全帽检测

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

跌倒检测

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

计划

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