ZhengPeng7
commited on
Commit
•
ec88fc8
1
Parent(s):
6a6592d
Move all BiRefNet github codes to the first level directory.
Browse files- birefnet.py +9 -9
birefnet.py
CHANGED
@@ -4,15 +4,15 @@ import torch.nn.functional as F
|
|
4 |
from kornia.filters import laplacian
|
5 |
from huggingface_hub import PyTorchModelHubMixin
|
6 |
|
7 |
-
from
|
8 |
-
from
|
9 |
-
from
|
10 |
-
from
|
11 |
-
from
|
12 |
-
from
|
13 |
-
from
|
14 |
-
from
|
15 |
-
from
|
16 |
|
17 |
|
18 |
class BiRefNet(
|
|
|
4 |
from kornia.filters import laplacian
|
5 |
from huggingface_hub import PyTorchModelHubMixin
|
6 |
|
7 |
+
from config import Config
|
8 |
+
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.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
|
16 |
|
17 |
|
18 |
class BiRefNet(
|