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": [
"SegFormerForImageSegmentation"
"SegformerForImageSegmentation"
],
"attention_probs_dropout_prob": 0.0,
"classifier_dropout_prob": 0.1,
@ -354,6 +354,7 @@
3,
3
],
"reshape_last_stage": true,
"sr_ratios": [
8,
4,
@ -367,5 +368,5 @@
2
],
"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_random_crop": false,
"do_resize": true,
"feature_extractor_type": "SegFormerFeatureExtractor",
"feature_extractor_type": "SegformerFeatureExtractor",
"image_mean": [
0.485,
0.456,