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.
|
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
|
|
|
|
|
|
|
|