Tim77777767 commited on
Commit
e9318b0
·
1 Parent(s): c93e685

HF Transformers Files removed

Browse files
README.md CHANGED
@@ -71,7 +71,7 @@ The weights of the Segformer (Original) model were used to get the inference res
71
 
72
  **Step 0.** Prerequisites
73
 
74
- - Pytorch: 2.3 (CUDA 12.1) (older versions should also work fine)
75
 
76
  **Step 1.** Clone Repository
77
 
 
71
 
72
  **Step 0.** Prerequisites
73
 
74
+ - Pytorch: 2.0.1 (CUDA 12.1) (older versions should also work fine)
75
 
76
  **Step 1.** Clone Repository
77
 
segformer_plusplus/config.json DELETED
@@ -1,10 +0,0 @@
1
- {
2
- "model_type": "segformerplusplus",
3
- "architectures": ["SegFormerPlusPlus"],
4
- "backbone": "b5",
5
- "supported_backbones": ["b0", "b1", "b2", "b3", "b4", "b5"],
6
- "head": "bsm_hq",
7
- "supported_heads": ["bsm_hq", "bsm_fast", "n2d_2x2"],
8
- "out_channels": 19,
9
- "num_labels": 19
10
- }
 
 
 
 
 
 
 
 
 
 
 
segformer_plusplus/configuration_segformer_plusplus.py DELETED
@@ -1,10 +0,0 @@
1
- from transformers import PretrainedConfig
2
-
3
- class SegformerPlusPlusConfig(PretrainedConfig):
4
- model_type = "segformer_plus_plus"
5
-
6
- def __init__(self, backbone_cfg=None, decode_head_cfg=None, out_channels=19, **kwargs):
7
- super().__init__(**kwargs)
8
- self.backbone_cfg = backbone_cfg
9
- self.decode_head_cfg = decode_head_cfg
10
- self.out_channels = out_channels
 
 
 
 
 
 
 
 
 
 
 
segformer_plusplus/modeling_segformer_plusplus.py DELETED
@@ -1,13 +0,0 @@
1
- from transformers import PreTrainedModel
2
- from .configuration_segformer_plusplus import SegformerPlusPlusConfig
3
- from .build_model import SegFormer
4
-
5
- class SegformerPlusPlusModel(PreTrainedModel):
6
- config_class = SegformerPlusPlusConfig
7
-
8
- def __init__(self, config: SegformerPlusPlusConfig):
9
- super().__init__(config)
10
- self.model = SegFormer(config)
11
-
12
- def forward(self, x, **kwargs):
13
- return self.model(x)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
segformer_plusplus/pytorch_model.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:2e064fa2fb7d618208c2542e76c543b7cb552a3d8997a0c6c4cc0a14da86ba58
3
- size 328287002