GengDavid/pytorch-cpn

About the utils/imutils.py line:41

lucas946 opened this issue · 1 comments

I think the following code is confusing or not correct,
heatmap /= am / 255
because batchnorm is the last layer of the predict net, the single element of heatmap should within the range of 0-1. I think the code should be correct as followed,
heatmap /= am.
But I'm not totally sure I am right, can you explain it?

Hope this would help you to understand.