fire-smoke-detect-yolov4 icon indicating copy to clipboard operation
fire-smoke-detect-yolov4 copied to clipboard

yolov5s fire-smoke weight is not the best!

Open gengyanlei opened this issue 3 years ago • 5 comments

注释: yolov5s 烟火检测模型是我随便训练几个epoch的结果,并不是最好的结果。目前只能开源1个看起来还说的过去的模型。最好的模型不会开源的。

gengyanlei avatar Nov 06 '20 06:11 gengyanlei

谢谢您,您应该不是使用的初始版本的yolov5和yolov4训练的吧,是有什么op的改进吗?我现在使用您的数据yolov5训练出来的结果中map=0.91 ,但是发现是看形状检测火焰的,误识别的情况还是挺高的,(其中数据是使用的您的模型预标,然后我修改,测试时候conf=0.7)

yutao007 avatar Feb 07 '21 08:02 yutao007

@yutao007 我用的yolov5 v3版本。yolov4无论怎么样 都可以读取测试的

gengyanlei avatar Feb 07 '21 09:02 gengyanlei

谢谢。我这边借用您有标签2000多张数据,拿您的模型预标无标签的百度数据,最后总共挑选了7000张数据,6400训练,600测试。我使用的是yolov5s 2版本的416*416,我发现我的火焰检测误检测很严重,用手比个三角形都会误检查。会不会是我的训练尺寸小了,或者是我的标签数据少了?

yutao007 avatar Feb 07 '21 12:02 yutao007

@yutao007 不是很清楚啊,特别小的、特别大的都不是很好预测

gengyanlei avatar Feb 08 '21 01:02 gengyanlei

恩拉,谢谢。我在看看怎么弄一下

yutao007 avatar Feb 08 '21 06:02 yutao007