CenterNet icon indicating copy to clipboard operation
CenterNet copied to clipboard

wh_loss is too large

Open unlabeledData opened this issue 5 years ago • 8 comments

Hi, I implemented the centernet on TF and used my own backbone net.

Then I found that when I trained my net on Pascal VOC 07+12, the hm_loss and off_loss decreased normally but wh_loss was oscillating around 20. ( hm_loss weight=1.0, wh_loss weight=0.1, off_loss weight=1.0)

Could you share some suggestions about how to set loss weights and how to tune the network to make it converge?

image

unlabeledData avatar Sep 05 '19 03:09 unlabeledData

I will suggest using our backbone network and fully reproduce our results first. This makes sure all details are correctly implemented.

xingyizhou avatar Sep 06 '19 00:09 xingyizhou

I have the same question when I just changed the backbone to a smaller and not pretrained backbone.

fuqichen1998 avatar Mar 31 '20 15:03 fuqichen1998

Hi,

Does the initialization of IDAUp matters while training CenterNet?

I re-implemented one model started training from scratch. The loss-hm plateaus to 9.208 and the network stopped converging. To be specific, the negative part of the heatmap focal loss ends up all zero.

But when I initialized the model with your implementation, the network trained normally.

How do you suggest debugging it?

ShawnNew avatar Jul 01 '20 09:07 ShawnNew

Hi,

Does the initialization of IDAUp matters while training CenterNet?

I re-implemented one model started training from scratch. The loss-hm plateaus to 9.208 and the network stopped converging. To be specific, the negative part of the heatmap focal loss ends up all zero.

But when I initialized the model with your implementation, the network trained normally.

How do you suggest debugging it?

I had the same problem. Do u have solve it?

Hello-Felix avatar Dec 17 '20 03:12 Hello-Felix

I had the same problem. I found that the boxmode of coco datasets is XYWH_abs. But the boxmode I set was XYXY_abs. I had solved the problem.

G-Apple1 avatar Feb 14 '22 13:02 G-Apple1

Hi @G-Apple1 can you elaborate on your solution?

vivcheng01 avatar Apr 24 '23 07:04 vivcheng01

Hi @G-Apple1 can you elaborate on your solution?

抱歉,我直接用中文回复。我当时的情况是生成coco数据集的时候出了问题,我用labelimg标注的软件是xyxy(左上角和右下角坐标),转成coco的时候是用的xywh(我以为是中心点坐标和宽高),实际上coco的是左上角坐标和宽高,改过来就没事了。

G-Apple1 avatar Apr 24 '23 07:04 G-Apple1

OK,谢谢!

vivcheng01 avatar Apr 26 '23 06:04 vivcheng01