the Resolution is lower

[brightendavid]

trafficstars

when I test your code ,I find the Resolution of output is lower than the input.How can I solve the problem.

[CauchyComplete]

Hi,

Heatmaps use a smaller size output, but you can use the pred variable in tools/infer.py to handle the full-resolution output.

https://github.com/mjkwon2021/CAT-Net/blob/17d294d82f9e402484bf82637ba992fdc304c0e7/tools/infer.py#L135

[brightendavid]

thank you for your answer, the resoluton of the output will always be (w//4,h//4),is that designed on porpose? or it may be a problem result from my torch version.

[CauchyComplete]

@brightendavid Yes, the outputs have 1/4 resolution. This is due to two stride-2 convolutions at the very beginning, as explained in the paper 😊