Jonathancasjar commited on
Commit
9b577de
1 Parent(s): 2f7ecca

Configurate json to run the model

Browse files
Files changed (1) hide show
  1. config.json +8 -32
config.json CHANGED
@@ -1,39 +1,15 @@
1
  {
2
  "_name_or_path": "Jonathancasjar/Retail_Shelves",
3
  "architectures": [
4
- "YolosForObjectDetection"
5
  ],
6
- "attention_probs_dropout_prob": 0.0,
7
- "auxiliary_loss": false,
8
- "bbox_cost": 5,
9
- "bbox_loss_coefficient": 5,
10
- "class_cost": 1,
11
- "eos_coefficient": 0.1,
12
- "giou_cost": 2,
13
- "giou_loss_coefficient": 2,
14
- "hidden_act": "gelu",
15
- "hidden_dropout_prob": 0.0,
16
- "hidden_size": 384,
17
  "id2label": {
18
- "LABEL_0": "missing",
19
- "LABEL_1": "product"
20
  },
21
- "image_size": [
22
- 512,
23
- 864
24
- ],
25
- "initializer_range": 0.02,
26
- "intermediate_size": 1536,
27
- "layer_norm_eps": 1e-12,
28
  "model_type": "bestv2.bin",
29
- "num_attention_heads": 6,
30
- "num_channels": 3,
31
- "num_detection_tokens": 100,
32
- "num_hidden_layers": 12,
33
- "patch_size": 16,
34
- "qkv_bias": true,
35
- "torch_dtype": "float32",
36
- "transformers_version": "4.31.0.dev0",
37
- "use_mid_position_embeddings": true
38
- }
39
-
 
1
  {
2
  "_name_or_path": "Jonathancasjar/Retail_Shelves",
3
  "architectures": [
4
+ "DetrForObjectDetection"
5
  ],
6
+ "dilation": true,
7
+ "dropout": 0.1,
8
+ "hidden_size": 256,
 
 
 
 
 
 
 
 
9
  "id2label": {
10
+ "0": "missing",
11
+ "1": "product"
12
  },
 
 
 
 
 
 
 
13
  "model_type": "bestv2.bin",
14
+ "num_labels": 5
15
+ }