Hi,
I have some questions about the pre-trained model keypointrcnn_resnet50_fpn from torchvision. I would like to do finetuning on my own dataset with 12 Keypoints to do single person pose keypoints detection. I’ve followed some code provided by the official documentation here 40, but not able to implement my own dataset and training loop because the example is mostly about segmentation.
My dataset only contains keypoints without information about visibility, so my first question is, can I train on this kind of data?
UPDATE: My code and error is in the next comment
Thank you,
Medhy