praeclarumjj3 commited on
Commit
8f8f5f5
1 Parent(s): d3d8281

Add Preprocessor Config

Browse files
Files changed (2) hide show
  1. config.json +12 -1
  2. preprocessor_config.json +20 -0
config.json CHANGED
@@ -18,7 +18,14 @@
18
  "oversample_ratio": 3.0,
19
  "importance_sample_ratio": 0.75,
20
  "init_std": 0.02,
21
- "init_xavier_std": 1.0
 
 
 
 
 
 
 
22
  },
23
  "inference_config": {
24
  "obj_mask_threshold": 0.8,
@@ -27,6 +34,9 @@
27
  },
28
  "backbone_config": {
29
  "image_size": 224,
 
 
 
30
  "patch_size": 4,
31
  "embed_dim": 96,
32
  "depths": [2,2,6,2],
@@ -57,6 +67,7 @@
57
  "hidden_dim": 256,
58
  "norm": "GN",
59
  "encoder_layers": 6,
 
60
  "decoder_layers": 10,
61
  "use_task_norm": true,
62
  "num_heads": 8,
 
18
  "oversample_ratio": 3.0,
19
  "importance_sample_ratio": 0.75,
20
  "init_std": 0.02,
21
+ "init_xavier_std": 1.0,
22
+ "layer_norm_eps": 1e-05,
23
+ "is_train": false,
24
+ "use_auxiliary_loss": true,
25
+ "output_attentions": false,
26
+ "output_hidden_states": false,
27
+ "use_return_dict": false,
28
+ "output_auxiliary_logits": false
29
  },
30
  "inference_config": {
31
  "obj_mask_threshold": 0.8,
 
34
  },
35
  "backbone_config": {
36
  "image_size": 224,
37
+ "num_channels": 3,
38
+ "hidden_act": "gelu",
39
+ "mlp_ratio": 4.0,
40
  "patch_size": 4,
41
  "embed_dim": 96,
42
  "depths": [2,2,6,2],
 
67
  "hidden_dim": 256,
68
  "norm": "GN",
69
  "encoder_layers": 6,
70
+ "encoder_feedforward_dim": 1024,
71
  "decoder_layers": 10,
72
  "use_task_norm": true,
73
  "num_heads": 8,
preprocessor_config.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "OneFormerFeatureExtractor",
5
+ "ignore_index": 255,
6
+ "image_mean": [
7
+ 0.48500001430511475,
8
+ 0.4560000002384186,
9
+ 0.4059999883174896
10
+ ],
11
+ "image_std": [
12
+ 0.2290000021457672,
13
+ 0.2239999920129776,
14
+ 0.22499999403953552
15
+ ],
16
+ "max_size": 2048,
17
+ "size": 512,
18
+ "size_divisibility": 32
19
+ }
20
+