ZhengPeng7 commited on
Commit
6a6592d
1 Parent(s): 108ae46

Move all BiRefNet github codes to the first level directory.

Browse files
birefnet.py CHANGED
@@ -9,7 +9,7 @@ from .dataset import class_labels_TR_sorted
9
  from .models.backbones.build_backbone import build_backbone
10
  from .models.modules.decoder_blocks import BasicDecBlk, ResBlk, HierarAttDecBlk
11
  from .models.modules.lateral_blocks import BasicLatBlk
12
- from .models.modules.aspp import ASPP, ASPPDeformable
13
  from .models.modules.ing import *
14
  from .models.refinement.refiner import Refiner, RefinerPVTInChannels4, RefUNet
15
  from .models.refinement.stem_layer import StemLayer
 
9
  from .models.backbones.build_backbone import build_backbone
10
  from .models.modules.decoder_blocks import BasicDecBlk, ResBlk, HierarAttDecBlk
11
  from .models.modules.lateral_blocks import BasicLatBlk
12
+ from .models.aspp import ASPP, ASPPDeformable
13
  from .models.modules.ing import *
14
  from .models.refinement.refiner import Refiner, RefinerPVTInChannels4, RefUNet
15
  from .models.refinement.stem_layer import StemLayer
models/{modules/aspp.py → aspp.py} RENAMED
File without changes
models/modules/refinement/__init__.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ from os.path import dirname, basename, isfile, join
2
+ import glob
3
+
4
+
5
+ modules = glob.glob(join(dirname(__file__), "*.py"))
6
+ __all__ = [ basename(f)[:-3] for f in modules if isfile(f) and not f.endswith('__init__.py')]