millionhz commited on
Commit
cc399b7
·
1 Parent(s): 39744b0

add preprocessor config

Browse files
Files changed (2) hide show
  1. config.json +2 -2
  2. preprocessor_config.json +18 -0
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "nvidia/mit-b0",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
@@ -68,4 +68,4 @@
68
  ],
69
  "torch_dtype": "float32",
70
  "transformers_version": "4.29.1"
71
- }
 
1
  {
2
+ "_name_or_path": "millionhz/segformer-b0-finetuned-flame",
3
  "architectures": [
4
  "SegformerForSemanticSegmentation"
5
  ],
 
68
  ],
69
  "torch_dtype": "float32",
70
  "transformers_version": "4.29.1"
71
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "SegformerFeatureExtractor",
5
+ "image_mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "image_std": [
11
+ 0.229,
12
+ 0.224,
13
+ 0.225
14
+ ],
15
+ "reduce_labels": true,
16
+ "resample": 2,
17
+ "size": 512
18
+ }