I modified the heatmap head to predict the heatmaps of 15 joint points. I use MSE loss.
Loss can converge to about 0.0002 on the training set, and about 0.001 on the verification set.
but after I visualized the prediction results of heatmaps, I found that the visual result is not good. I don't know why?
Please consult everyone!
heatmap GT:
heatmap pred:
I modified the heatmap head to predict the heatmaps of 15 joint points. I use MSE loss. Loss can converge to about 0.0002 on the training set, and about 0.001 on the verification set. but after I visualized the prediction results of heatmaps, I found that the visual result is not good. I don't know why? Please consult everyone! heatmap GT: heatmap pred: