Jimmy_2times:

I managed to do transfer learning on a ResNet-18 model with my custom dataset for object detection. It seems quite straight forward with Pytorch.


seems like you already have some code?
either ways, feel free to use something like https://pytorch.org/hub/pytorch_vision_deeplabv3_resnet101/ 37
there might be some samples at https://github.com/NVIDIA-AI-IOT 22