improve stability of data processing

dk-liang / FIDTM

[IEEE TMM] Focal Inverse Distance Transform Maps for Crowd Localization

MIT License

169 stars 41 forks source link

Closed rydenisbak closed 1 year ago

rydenisbak commented 1 year ago

Hello! Thanks for the cool work.

add padding (right and bottom) if image size is less than crop size
make contiguous numpy arrays after flip (my torch 1.12 throws exception for non contiguous arrays)