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

 
 
 

Contributors