TencentYoutuResearch / CrowdCounting-P2PNet

The official codes for the ICCV2021 Oral presentation "Rethinking Counting and Localization in Crowds: A Purely Point-Based Framework"
Other
423 stars 114 forks source link

Hardcoded pyramid_levels causing runtime error with stride customization #71

Open hanoglu opened 7 months ago

hanoglu commented 7 months ago

There is no option in the training code to change the stride variable. Apparently, stride depends on pyramid_levels value. But this value is hardcoded in the line self.anchor_points = AnchorPoints(pyramid_levels=[3,], row=row, line=line) and in case it is changed, it gives this runtime error; RuntimeError: The size of tensor a (1024) must match the size of tensor b (256) at non-singleton dimension at output_coord = regression + anchor_points