Train custom dataset with fasterrcnn_resnet50_fpn model

TorchStudio / torchstudio

IDE for PyTorch and its ecosystem

https://torchstudio.ai

MIT License

378 stars 27 forks source link

Train custom dataset with fasterrcnn_resnet50_fpn model #43

Open pbanerjee0 opened 1 year ago

pbanerjee0 commented 1 year ago

I am able to load the custom dataset where target contain dict of {boxes, labels}. But, when I try to build fasterrcnn_resnet50_fpn model, it is showing TorchStudio\python\lib\site-packages\torch__init__.py, line 853, in _assert assert condition, message AssertionError: targets should not be none when in training mode error.

Please help me resolve this issue. @divideconcept

pbanerjee0 commented 1 year ago

Hi @divideconcept ,

Could you please help me to solve the above issue?