How to train own model and annotations for this?
sharon-samson opened this issue · comments
Hi Iam currently working on a research project, and it will be helpful if some one explain how can we train our own model and how to create its annotation files..
@sharon-samson Hi, I don't know if I'm too late or not but I think that this tutorial in PyTorch is helpful. Just create your own dataset class and go on with the tutorial, it applies Mask-RCNN which should do a decent job in an instance segmentation task ( it will also output the bounding boxes and their labels ).
https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html
I got segmented output. How can I make landmark points on it?