Move all BiRefNet github codes to the first level directory.

models/aspp.py

@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from models.modules.deform_conv import DeformableConv2d
+from models.deform_conv import DeformableConv2d
 from config import Config
 
 

birefnet.py → models/birefnet.py

@@ -6,13 +6,13 @@ from huggingface_hub import PyTorchModelHubMixin
 
 from config import Config
 from dataset import class_labels_TR_sorted
-from models.backbones.build_backbone import build_backbone
-from models.modules.decoder_blocks import BasicDecBlk, ResBlk, HierarAttDecBlk
-from models.modules.lateral_blocks import BasicLatBlk
+from models.build_backbone import build_backbone
+from models.decoder_blocks import BasicDecBlk, ResBlk, HierarAttDecBlk
+from models.lateral_blocks import BasicLatBlk
 from models.aspp import ASPP, ASPPDeformable
-from models.modules.ing import *
-from models.refinement.refiner import Refiner, RefinerPVTInChannels4, RefUNet
-from models.refinement.stem_layer import StemLayer
+from models.ing import *
+from models.refiner import Refiner, RefinerPVTInChannels4, RefUNet
+from models.stem_layer import StemLayer
 
 
 class BiRefNet(

models/{backbones/build_backbone.py → build_backbone.py}

@@ -2,8 +2,8 @@ import torch
 import torch.nn as nn
 from collections import OrderedDict
 from torchvision.models import vgg16, vgg16_bn, VGG16_Weights, VGG16_BN_Weights, resnet50, ResNet50_Weights
-from models.backbones.pvt_v2 import pvt_v2_b0, pvt_v2_b1, pvt_v2_b2, pvt_v2_b5
-from models.backbones.swin_v1 import swin_v1_t, swin_v1_s, swin_v1_b, swin_v1_l
+from models.pvt_v2 import pvt_v2_b0, pvt_v2_b1, pvt_v2_b2, pvt_v2_b5
+from models.swin_v1 import swin_v1_t, swin_v1_s, swin_v1_b, swin_v1_l
 from config import Config
 
 

models/{modules/decoder_blocks.py → decoder_blocks.py}

@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
-from models.modules.aspp import ASPP, ASPPDeformable
-from models.modules.attentions import PSA, SGE
+from models.aspp import ASPP, ASPPDeformable
+from models.attentions import PSA, SGE
 from config import Config
 
 

models/{modules/ing.py → ing.py}

@@ -1,5 +1,5 @@
 import torch.nn as nn
-from models.modules.mlp import MLPLayer
+from models.mlp import MLPLayer
 
 
 class BlockA(nn.Module):

models/modules/refinement/refiner.py

@@ -9,11 +9,11 @@ from torchvision.models import resnet50
 
 from config import Config
 from dataset import class_labels_TR_sorted
-from models.backbones.build_backbone import build_backbone
-from models.modules.decoder_blocks import BasicDecBlk
-from models.modules.lateral_blocks import BasicLatBlk
-from models.modules.ing import *
-from models.refinement.stem_layer import StemLayer
+from models.build_backbone import build_backbone
+from models.decoder_blocks import BasicDecBlk
+from models.lateral_blocks import BasicLatBlk
+from models.ing import *
+from models.stem_layer import StemLayer
 
 
 class RefinerPVTInChannels4(nn.Module):

models/modules/refinement/stem_layer.py

@@ -1,5 +1,5 @@
 import torch.nn as nn
-from models.modules.utils import build_act_layer, build_norm_layer
+from models.utils import build_act_layer, build_norm_layer
 
 
 class StemLayer(nn.Module):

models/{refinement/refiner.py → refiner.py}

@@ -9,11 +9,11 @@ from torchvision.models import resnet50
 
 from config import Config
 from dataset import class_labels_TR_sorted
-from models.backbones.build_backbone import build_backbone
-from models.modules.decoder_blocks import BasicDecBlk
-from models.modules.lateral_blocks import BasicLatBlk
-from models.modules.ing import *
-from models.refinement.stem_layer import StemLayer
+from models.build_backbone import build_backbone
+from models.decoder_blocks import BasicDecBlk
+from models.lateral_blocks import BasicLatBlk
+from models.ing import *
+from models.stem_layer import StemLayer
 
 
 class RefinerPVTInChannels4(nn.Module):

models/{refinement/stem_layer.py → stem_layer.py}

@@ -1,5 +1,5 @@
 import torch.nn as nn
-from models.modules.utils import build_act_layer, build_norm_layer
+from models.utils import build_act_layer, build_norm_layer
 
 
 class StemLayer(nn.Module):