centernet-pytorch 使用自己的数据集，可以正常训练，但get_dr

使用自己的数据集，可以正常训练，但get_dr_txt.py使用时，得到的txt都为空白

Open WH1123 opened this issue 4 years ago • 3 comments

您好，我使用您的程序训练自己的数据集，使用的是resnet50，使用nms。按照步骤我已经成功的完成了训练阶段，得到了Epoch100-Total_Loss0.0002-Val_Loss0.0002.pth。使用这个权重去进行评估，在运行get_dr_txt.py程序后，在detection-results文件下，为每张图片创建了对应的txt文件，但这里得到的文件都是空白的。监控get_dr_txt.py程序，发现outputs=decode_bbox(...)输出的结果都为空的列表。我尝试解决，但不了解为什么预测不出结果，请您帮忙指导。

Jun 21 '21 08:06 WH1123

先看问题汇总，你应该没有训练的目标

Jun 22 '21 08:06 bubbliiiing

@WH1123 你好，我跟你遇到了同样问题，请问你是怎么解决的？

Oct 05 '21 01:10 ZNHzlq

检查了嘛

Oct 05 '21 16:10 bubbliiiing

centernet-pytorch centernet-pytorch copied to clipboard

使用自己的数据集，可以正常训练，但get_dr_txt.py使用时，得到的txt都为空白

centernet-pytorch
centernet-pytorch copied to clipboard