Skip to content

README.md

Latest commit

 

History

History
15 lines (8 loc) · 798 Bytes

README.md

File metadata and controls

15 lines (8 loc) · 798 Bytes

timmFasterRcnn

Use timm pretrained backbones for you FasterRCNN!

  1. model_config.py -> it returns the model,feat_sizes,output channel and the feat layer names, which is reqd by the Add_FPN.py file

  2. Add_FPN.py -> Edited the BackboneWithFPN function from pytorch, which is now used to add FPN to any timm model, till now it can be only used for Efficentnet family

  3. test.ipynb -> an example explaining how to use it any backbone and add FPN to it

Some problems

Loading even small models takes up huge amout of memory, I cant quite detect why, although the training and inferencing speed is maintained. (For example if we add effnet_b1 in the FasterRcnn, which is smaller in terms of params of Resnet101,which is a default option in FasterRcnn still it occupies larger space in memory)