Duankaiwen / CenterNet

Codes for our paper "CenterNet: Keypoint Triplets for Object Detection" .

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to change the input image size

cuongnc220592 opened this issue · comments

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.