Object detection models

[denisvlr]

Pytorch recently included pretrained object detection models such FastRCNN ( torchvision.models.detection.faster_rcnn). Do you know how to integrate them with fastai?

[PPPW]

Hi @denisvlr, PyTorch provided a very good tutorial on this topic: https://pytorch.org/tutorials/intermediate/torchvision_tutorial.html

It does the transfer learning, and we can customize the learning rate scheduling. If we want to use it with fastai, we have to customize the data loader as well as the loss function, as the PyTorch model requires both image and target as input, and the forward returns the loss during training. I don't think it's worthy tweaking it to fit the fastai framework, I think it's much less work by customizing from the above tutorial...