bcmi / FOPA-Fast-Object-Placement-Assessment

A discriminative object placement approach
MIT License
32 stars 1 forks source link

只有positive label,neg为空,训练出的heatmap几乎全白。 #9

Closed Xiangyu33 closed 4 days ago

Xiangyu33 commented 4 days ago

仿照原始的训练数据格式,整理了自己的训练数据。数据格式基本如下,只有pos_label,neg_label为空: image

训练过程loss正常收敛

这样训练后的推理图像,每个heatmap都几乎为全白 image 与原始作者训练结果有较大差距,如下: image

请问只有pos_label的训练,只有全白heatmap是否符合预期

WhynotHAHA commented 4 days ago

只有pos_label的训练,只会输出全白heatmap,因为只要全都预测为positive就符合训练集需求

Xiangyu33 commented 4 days ago

我试试其他加入neg的方式(随机高斯点、边界点、其他检测物的坐标),效果不好我再找人工试试