giddyyupp / coco-minitrain

a subset of coco dataset for faster experimentation
236 stars 34 forks source link

Result table for keypoint detection #32

Open kcs6568 opened 1 year ago

kcs6568 commented 1 year ago

Thank you for your excellent project!

I'm training the keypoint RCNN network implemented in torchvision by scaling the image to the size denoted in object detection result tables.

I know the size of input image between simplebaseline2D and networks of RCNN family(faster, mask, etc.) is different.

I'd like to know how to scale the input image size for minicoco dataset in the keypoint detection experiment using a simplebaseline2D network.

Despite using different networks, is the same size between the object detection task and the keypoint detection task?

Thanks in advance!

giddyyupp commented 1 year ago

Hello, we trained pose estimation models using MMPose. So for each method you could check the corresponding config file to see what the image size is to train the models.

For SimpleBaseline2D, we resized the image to 256x192 pixels for training.