训练100个epoch后验证时ap都为0,推理时基本都是错误的
sakamoto111 opened this issue · 0 comments
sakamoto111 commented
大佬您好,能否帮我看看。训练采用自己的数据集(200张),标注完并使用您提供的xml2txt.py转换为txt文件后,运行display_inputs.py检查训练集的图片没有问题。训练完后打开TensorBoard各项loss也基本收敛了,各类别的precision和recall都显示正常。但是运行test.py验证时AP均为0,以及detect.py推理时都是错误的杂框,比自己标注的gt要小很多。下载了您的trash数据集并训练10个epoch后ap就以及很高了,训练及验证时的参数也与您的基本一致,没有改动。
我感觉还是数据集转换的问题,但是找不到具体错在哪里,您方便帮我看一下吗,感谢!
mAP为0:
部分PR如下:
display_inputs.py转换结果:
推理错误框: