search

bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库，可以利用voc数据集格式的数据进行训练。
MIT License
1.52k stars 351 forks source link

训练时loss值很低，得到的模型形同虚设，检测不到目标 #206

Open maige1219 opened 1 month ago

maige1219 commented 1 month ago

我想训练一个人脸检测的模型，从roboflow上找了一个数据集后以VOC格式下载了（但是它私自加上了后缀名）。 image

然后一步一步按照作者的步骤来进行的，结果训练时loss的值很奇怪， image

而且使用得到的模型进行预测得不到任何结果，什么图进去，什么图出来。

望大佬相助

kkkkkkkb commented 5 days ago

我觉得是数据集的问题，你可以让chatgpt写个代码把这些文件的后缀都去掉，然后再重新划分重新训练试试，感觉名字有时候太冗余也会出各种问题。（活着你查看下train.txt 2007_train.txt 等文件里面的文件名格式、地址都是对的吗