Burdenthrive commited on
Commit
e3bc734
·
verified ·
1 Parent(s): fdd0e8b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -0
config.json CHANGED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "task": "image-segmentation",
3
+ "model_name": "segformer-mit-b4",
4
+ "model_kwargs": {
5
+ "encoder_name": "mit_b4",
6
+ "encoder_weights": "imagenet",
7
+ "in_channels": 3,
8
+ "num_classes": 4,
9
+ },
10
+ "classes": ["clear", "thick cloud", "thin cloud", "cloud shadow"],
11
+ "input_bands": ["B04", "B03", "B02"]
12
+ }