hf-transformers-bot commited on
Commit
771f716
·
verified ·
1 Parent(s): aac6213

Update tiny models for OmDetTurboForObjectDetection

Browse files
config.json ADDED
@@ -0,0 +1,124 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "apply_layernorm_after_vision_backbone": false,
3
+ "architectures": [
4
+ "OmDetTurboForObjectDetection"
5
+ ],
6
+ "backbone": null,
7
+ "backbone_config": {
8
+ "attention_probs_dropout_prob": 0.0,
9
+ "depths": [
10
+ 1,
11
+ 1,
12
+ 1,
13
+ 1
14
+ ],
15
+ "drop_path_rate": 0.1,
16
+ "embed_dim": 8,
17
+ "encoder_stride": 32,
18
+ "hidden_act": "gelu",
19
+ "hidden_dropout_prob": 0.0,
20
+ "hidden_size": 64,
21
+ "image_size": 224,
22
+ "initializer_range": 0.02,
23
+ "layer_norm_eps": 1e-05,
24
+ "mlp_ratio": 4.0,
25
+ "model_type": "swin",
26
+ "num_channels": 3,
27
+ "num_heads": [
28
+ 1,
29
+ 1,
30
+ 1,
31
+ 1
32
+ ],
33
+ "num_layers": 4,
34
+ "out_features": [
35
+ "stage2",
36
+ "stage3",
37
+ "stage4"
38
+ ],
39
+ "out_indices": [
40
+ 2,
41
+ 3,
42
+ 4
43
+ ],
44
+ "patch_size": 4,
45
+ "qkv_bias": true,
46
+ "stage_names": [
47
+ "stem",
48
+ "stage1",
49
+ "stage2",
50
+ "stage3",
51
+ "stage4"
52
+ ],
53
+ "use_absolute_embeddings": false,
54
+ "window_size": 7
55
+ },
56
+ "batch_norm_eps": 1e-05,
57
+ "cache_size": 100,
58
+ "class_distance_type": "cosine",
59
+ "class_embed_dim": 16,
60
+ "conv_norm_activation": "gelu",
61
+ "csp_activation": "silu",
62
+ "d_model": 8,
63
+ "decoder_activation": "relu",
64
+ "decoder_dim_feedforward": 32,
65
+ "decoder_dropout": 0.0,
66
+ "decoder_hidden_dim": 8,
67
+ "decoder_num_heads": 2,
68
+ "decoder_num_layers": 2,
69
+ "decoder_num_points": 4,
70
+ "disable_custom_kernels": false,
71
+ "dtype": "float32",
72
+ "encoder_attention_heads": 2,
73
+ "encoder_dim_feedforward": 32,
74
+ "encoder_dropout": 0.0,
75
+ "encoder_feedforward_activation": "relu",
76
+ "encoder_feedforward_dropout": 0.0,
77
+ "encoder_in_channels": [
78
+ 16,
79
+ 32,
80
+ 64
81
+ ],
82
+ "encoder_layers": 2,
83
+ "encoder_projection_indices": [
84
+ 2
85
+ ],
86
+ "eval_size": null,
87
+ "hidden_expansion": 1,
88
+ "image_size": 224,
89
+ "init_std": 0.02,
90
+ "is_encoder_decoder": true,
91
+ "layer_norm_eps": 1e-05,
92
+ "learn_initial_query": false,
93
+ "model_type": "omdet-turbo",
94
+ "num_feature_levels": 3,
95
+ "num_queries": 20,
96
+ "positional_encoding_temperature": 10000,
97
+ "task_encoder_hidden_dim": 32,
98
+ "text_config": {
99
+ "attention_dropout": 0.0,
100
+ "bos_token_id": 49406,
101
+ "eos_token_id": 49407,
102
+ "hidden_act": "quick_gelu",
103
+ "hidden_size": 16,
104
+ "initializer_factor": 1.0,
105
+ "initializer_range": 0.02,
106
+ "intermediate_size": 16,
107
+ "layer_norm_eps": 1e-05,
108
+ "max_position_embeddings": 200,
109
+ "model_type": "clip_text_model",
110
+ "num_attention_heads": 2,
111
+ "num_hidden_layers": 2,
112
+ "pad_token_id": 49407,
113
+ "projection_dim": 512,
114
+ "vocab_size": 49408
115
+ },
116
+ "text_projection_in_dim": 16,
117
+ "text_projection_out_dim": 16,
118
+ "transformers_version": "5.16.0.dev0",
119
+ "vision_features_channels": [
120
+ 8,
121
+ 8,
122
+ 8
123
+ ]
124
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f83383913569dbb1e917bdcfb8fc43066a7dccbd7ae15e43f4d1a23fc59490bb
3
+ size 3647208
preprocessor_config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_convert_annotations": true,
7
+ "do_normalize": true,
8
+ "do_pad": true,
9
+ "do_rescale": true,
10
+ "do_resize": true,
11
+ "format": "coco_detection",
12
+ "image_mean": [
13
+ 0.485,
14
+ 0.456,
15
+ 0.406
16
+ ],
17
+ "image_processor_type": "DetrImageProcessor",
18
+ "image_std": [
19
+ 0.229,
20
+ 0.224,
21
+ 0.225
22
+ ],
23
+ "resample": 2,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "size": {
26
+ "height": 224,
27
+ "width": 224
28
+ }
29
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|startoftext|>",
5
+ "clean_up_tokenization_spaces": true,
6
+ "do_lower_case": true,
7
+ "eos_token": "<|endoftext|>",
8
+ "errors": "replace",
9
+ "is_local": true,
10
+ "local_files_only": false,
11
+ "model_max_length": 77,
12
+ "pad_token": "<|endoftext|>",
13
+ "processor_class": "OmDetTurboProcessor",
14
+ "tokenizer_class": "CLIPTokenizer",
15
+ "unk_token": "<|endoftext|>"
16
+ }