edit the LRBranch and HRBranch after changed backbone

[luoww1992]

i prepare to change the backbone to use efficient,
while how to edit the func LRBranch and HRBranch ?

[munibkhanali]

HI @luoww1992
You can use timm https://github.com/huggingface/pytorch-image-models
add the following class in the wrapper.py

class Efficienb0v1(BaseBackbone):
    """ 
        EfficientBackbone 
    """

    def __init__(self, in_channels):
        super(Efficienb0v1, self).__init__(in_channels)

        self.model = timm.create_model('tf_efficientnet_b0.ns_jft_in1k',pretrained=True,features_only=True)
        self.enc_channels = self.model.feature_info.channels()

    def forward(self, x):
       return self.model.forward(x)

then you have to update backbone->init.py accodingly

I hope it will helps
Thank you

[luoww1992]

@munibkhanali

i use the efficient07 backbone from this project : https://github.com/Fire-friend/dugMatting/blob/main/models/backbones/wrapper.py#L578

how to edit ?

[munibkhanali]

HI @luoww1992
The code there is pretty straightforward you don't need to change anything in LRBranch or HRBranch. if you are facing any issues please describe them in detail
Thank you.