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.
|
| 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')]
|