Skip to content

mrinath123/timmFasterRcnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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)

About

Adds timm pretrained backbone to pytorch's FasterRcnn model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published