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

Move all BiRefNet github codes to the first level directory.

Browse files
Files changed (1) hide show
  1. 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 .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(
 
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(