mhassanch commited on
Commit
3739afa
·
1 Parent(s): c7ab3da

add pre processor config and update model config

Browse files
Files changed (2) hide show
  1. config.json +17 -1
  2. preprocessor_config.json +7 -0
config.json CHANGED
@@ -1,3 +1,19 @@
1
  {
2
- "model_type" : ""
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  }
 
1
  {
2
+ "classes": [
3
+ "plane",
4
+ "baseball-diamond",
5
+ "bridge",
6
+ "ground-track-field",
7
+ "small-vehicle",
8
+ "large-vehicle",
9
+ "ship",
10
+ "tennis-court",
11
+ "basketball-court",
12
+ "storage-tank",
13
+ "soccer-ball-field",
14
+ "roundabout",
15
+ "harbor",
16
+ "swimming-pool",
17
+ "helicopter"
18
+ ]
19
  }
preprocessor_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_format": "channels_first",
3
+ "default_to_square": true,
4
+ "device": null,
5
+ "do_convert_rgb": true,
6
+ "return_tensors": true
7
+ }