naver / r2d2

Other
461 stars 86 forks source link

Fix crash when the input image is small #49

Open sarlinpe opened 1 year ago

sarlinpe commented 1 year ago

When max(H, W) < min_size, the inference is skipped and the code crashes at torch.cat(Y).