training failed in SRT heatmaps？

[Dian-Yi]

training failed in SRT heatmaps？

[Dian-Yi]

I did some experiments to find problems.
There is a sample heatmap label data from: def generate_label_map(pts, height, width, sigma, downsample, nopoints, ctype):
the sum of each map's value > 0, sigma:4, downsample:8:
[ 7, 6, 8, 7, 7, 7, 7, 7, 7, 8, 7, 8,
7, 7, 7, 7, 8, 8, 7, 8, 7, 7, 8, 8,
8, 7, 8, 7, 7, 7, 7, 8, 7, 7, 7, 7,
7, 7, 7, 8, 7, 7, 7, 7, 7, 8, 6, 8,
7, 7, 8, 7, 8, 8, 7, 7, 6, 7, 7, 8,
8, 8, 7, 8, 7, 7, 8, 7, 1024]
the max value of each maps:
[0.8979, 0.8032, 0.6506, 0.6165, 0.6372, 0.4791, 0.5207, 0.4606, 0.8633,
0.8925, 0.8269, 0.7173, 0.8287, 0.5979, 0.5846, 0.6472, 0.7989, 0.7900,
0.9217, 0.6162, 0.6745, 0.5610, 0.8472, 0.8779, 0.0000, 0.9160, 0.5829,
0.4558, 0.6468, 0.4131, 0.6791, 0.7397, 0.8015, 0.6352, 0.7229, 0.6346,
0.4943, 0.6630, 0.8280, 0.6312, 0.9719, 0.9219, 0.7392, 0.7575, 0.6396,
0.9408, 0.8714, 0.8833, 0.6204, 0.6979, 0.6662, 0.8334, 0.7105, 0.7566,
0.9396, 0.8781, 0.6446, 0.9268, 0.7921, 0.9549, 0.5869, 0.9428, 0.8311,
0.9922, 0.5339, 0.8815, 0.9315, 0.9609, 1.0000]

you can find the number of background label is 32*32. There is an imbalance between background and foreground. so i change the function(generate_label_map) and add AwingLoss and loss weight maps. They are all from a paper(AwingLoss). After,my model can be trianed well.
Have you ever encountered such a problem when you train the ProCpm model ?