Tim77777767
commited on
Commit
·
245b43a
1
Parent(s):
e9318b0
README modified
Browse files- README.md +0 -1
- segformer-cityscapes-b5-bsm-hq.py +14 -0
README.md
CHANGED
@@ -99,7 +99,6 @@ python3 -m segformer_plusplus.start_cityscape_benchmark --backbone [b1-b5] --hea
|
|
99 |
```
|
100 |
|
101 |
Checkpoints can be downloaded via the provided links above:
|
102 |
-
Example:
|
103 |
```shell
|
104 |
wget -O [NameOfDownloadedFile] "URL of Model-Download"
|
105 |
```
|
|
|
99 |
```
|
100 |
|
101 |
Checkpoints can be downloaded via the provided links above:
|
|
|
102 |
```shell
|
103 |
wget -O [NameOfDownloadedFile] "URL of Model-Download"
|
104 |
```
|
segformer-cityscapes-b5-bsm-hq.py
ADDED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
_base_ = ['../../../configs/segformer/segformer_mit-b5_8xb1-160k_cityscapes-1024x1024.py']
|
2 |
+
|
3 |
+
model = dict(
|
4 |
+
backbone=dict(
|
5 |
+
tome_cfg=[
|
6 |
+
dict(kv_mode='bsm', kv_r=0.6, kv_sx=2, kv_sy=2),
|
7 |
+
dict(kv_mode='bsm', kv_r=0.6, kv_sx=2, kv_sy=2),
|
8 |
+
dict(q_mode='bsm', q_r=0.8, q_sx=4, q_sy=4),
|
9 |
+
dict(q_mode='bsm', q_r=0.8, q_sx=4, q_sy=4)
|
10 |
+
]
|
11 |
+
)
|
12 |
+
)
|
13 |
+
|
14 |
+
train_dataloader = dict(batch_size=4, num_workers=8)
|