nvidia
/
segformer-b0-finetuned-ade-512-512
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update model

This commit is contained in:
Niels Rogge 2021-10-15 08:42:13 +00:00
parent 409aa62386
commit 77fa31aeb8
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
config.json
View File

@ -1,6 +1,6 @@
{ {
"architectures": [ "architectures": [
"SegFormerForImageSegmentation" "SegformerForImageSegmentation"
], ],
"attention_probs_dropout_prob": 0.0, "attention_probs_dropout_prob": 0.0,
"classifier_dropout_prob": 0.1, "classifier_dropout_prob": 0.1,
@ -354,6 +354,7 @@
3, 3,
3 3
], ],
"reshape_last_stage": true,
"sr_ratios": [ "sr_ratios": [
8, 8,
4, 4,
@ -367,5 +368,5 @@
2 2
], ],
"torch_dtype": "float32", "torch_dtype": "float32",
"transformers_version": "4.10.0.dev0" "transformers_version": "4.12.0.dev0"
} }

2
preprocessor_config.json
View File

@ -8,7 +8,7 @@
"do_pad": true, "do_pad": true,
"do_random_crop": false, "do_random_crop": false,
"do_resize": true, "do_resize": true,
"feature_extractor_type": "SegFormerFeatureExtractor", "feature_extractor_type": "SegformerFeatureExtractor",
"image_mean": [ "image_mean": [
0.485, 0.485,
0.456, 0.456,