update
Browse files- app.py +1 -1
- configs/_base_/models/upernet_uniformer.py +1 -1
app.py
CHANGED
@@ -14,7 +14,7 @@ from huggingface_hub import hf_hub_download
|
|
14 |
# Set to cuda to load on GPU
|
15 |
device = "cpu"
|
16 |
checkpoint_file = hf_hub_download(repo_id="Andy1621/uniformer", filename="upernet_global_small.pth")
|
17 |
-
config_file = './exp/
|
18 |
# init detector
|
19 |
# build the model from a config file and a checkpoint file
|
20 |
model = init_segmentor(config_file, checkpoint_file, device='cpu')
|
|
|
14 |
# Set to cuda to load on GPU
|
15 |
device = "cpu"
|
16 |
checkpoint_file = hf_hub_download(repo_id="Andy1621/uniformer", filename="upernet_global_small.pth")
|
17 |
+
config_file = './exp/upernet_global_small/test_config_h32.py'
|
18 |
# init detector
|
19 |
# build the model from a config file and a checkpoint file
|
20 |
model = init_segmentor(config_file, checkpoint_file, device='cpu')
|
configs/_base_/models/upernet_uniformer.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
# model settings
|
2 |
-
norm_cfg = dict(type='
|
3 |
model = dict(
|
4 |
type='EncoderDecoder',
|
5 |
pretrained=None,
|
|
|
1 |
# model settings
|
2 |
+
norm_cfg = dict(type='BN', requires_grad=True)
|
3 |
model = dict(
|
4 |
type='EncoderDecoder',
|
5 |
pretrained=None,
|