Duankaiwen / CenterNet

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .
MIT License
1.86k stars 384 forks source link

How to change the input image size #130

Open cuongnc220592 opened 4 years ago

cuongnc220592 commented 4 years ago

I have a question for changing the size of input image. As the configuration, the input size is 511x511. However, I want to resize the input image to 1024x768 or 1024x1024. I have tried to change the config file "input_size": [1023, 1023].

But I have a error IndexError: The shape of the mask [1, 80, 128, 128] at index 2 does not match the shape of the indexed tensor [1, 80, 256, 256] at index 2

How can I solve the problem Thank you very much.