修改结构

birefnet.py

@@ -5,7 +5,7 @@
 import torch.nn.functional as F
 from PIL import Image
 from models.baseline import BiRefNet
-from config import Config
+from bf.config import Config
 from torchvision.transforms.functional import normalize
 import numpy as np
 import folder_paths

dataset.py

@@ -6,8 +6,8 @@
 from torchvision import transforms
 
 from preproc import preproc
-from config import Config
-from utils import path_to_image
+from bf.config import Config
+from bf.utils import path_to_image
 
 
 Image.MAX_IMAGE_PIXELS = None       # remove DecompressionBombWarning

models/backbones/build_backbone.py

@@ -4,7 +4,7 @@
 from torchvision.models import vgg16, vgg16_bn, VGG16_Weights, VGG16_BN_Weights, resnet50, ResNet50_Weights
 from models.backbones.pvt_v2 import 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 config import Config
+from bf.config import Config
 
 
 config = Config()

models/backbones/pvt_v2.py

@@ -7,7 +7,7 @@
 
 import math
 
-from config import Config
+from bf.config import Config
 
 config = Config()
 

models/backbones/swin_v1.py

@@ -12,7 +12,7 @@
 import numpy as np
 from timm.models.layers import DropPath, to_2tuple, trunc_normal_
 
-from config import Config
+from bf.config import Config
 
 
 config = Config()

models/baseline.py

@@ -16,7 +16,7 @@
 from models.refinement.refiner import Refiner, RefinerPVTInChannels4, RefUNet
 from models.refinement.stem_layer import StemLayer
 
-from config import Config
+from bf.config import Config
 from dataset import class_labels_TR_sorted
 
 

models/models/common.py

@@ -22,7 +22,7 @@
 from PIL import Image
 from torch.cuda import amp
 
-from utils import TryExcept
+from bf.utils import TryExcept
 from utils.dataloaders import exif_transpose, letterbox
 from utils.general import (LOGGER, ROOT, Profile, check_requirements, check_suffix, check_version, colorstr,
                            increment_path, is_notebook, make_divisible, non_max_suppression, scale_boxes,

models/modules/aspp.py

@@ -2,7 +2,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 from models.modules.deform_conv import DeformableConv2d
-from config import Config
+from bf.config import Config
 
 
 config = Config()

models/modules/decoder_blocks.py

@@ -2,7 +2,7 @@
 import torch.nn as nn
 from models.modules.aspp import ASPP, ASPPDeformable
 from models.modules.attentions import PSA, SGE
-from config import Config
+from bf.config import Config
 
 
 config = Config()

models/modules/lateral_blocks.py

@@ -4,7 +4,7 @@
 import torch.nn.functional as F
 from functools import partial
 
-from config import Config
+from bf.config import Config
 
 
 config = Config()

models/refinement/refiner.py

@@ -7,7 +7,7 @@
 from torchvision.models import vgg16, vgg16_bn
 from torchvision.models import resnet50
 
-from config import Config
+from bf.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