Felladrin commited on
Commit
592854e
·
verified ·
1 Parent(s): 98298cb

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers.js
3
+ base_model:
4
+ - nicky007/BalloonDetectioDTR
5
+ ---
6
+
7
+ # BalloonDetectioDTR (ONNX)
8
+
9
+ This is an ONNX version of [nicky007/BalloonDetectioDTR](https://huggingface.co/nicky007/BalloonDetectioDTR). It was automatically converted and uploaded using [this space](https://huggingface.co/spaces/onnx-community/convert-to-onnx).
config.json ADDED
@@ -0,0 +1,146 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "_name_or_path": "nicky007/BalloonDetectioDTR",
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "relu",
6
+ "architectures": [
7
+ "DetrForObjectDetection"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "auxiliary_loss": false,
11
+ "backbone": null,
12
+ "backbone_config": {
13
+ "_attn_implementation_autoset": false,
14
+ "_name_or_path": "",
15
+ "add_cross_attention": false,
16
+ "architectures": null,
17
+ "bad_words_ids": null,
18
+ "begin_suppress_tokens": null,
19
+ "bos_token_id": null,
20
+ "chunk_size_feed_forward": 0,
21
+ "cross_attention_hidden_size": null,
22
+ "decoder_start_token_id": null,
23
+ "depths": [
24
+ 3,
25
+ 4,
26
+ 6,
27
+ 3
28
+ ],
29
+ "diversity_penalty": 0.0,
30
+ "do_sample": false,
31
+ "downsample_in_bottleneck": false,
32
+ "downsample_in_first_stage": false,
33
+ "early_stopping": false,
34
+ "embedding_size": 64,
35
+ "encoder_no_repeat_ngram_size": 0,
36
+ "eos_token_id": null,
37
+ "exponential_decay_length_penalty": null,
38
+ "finetuning_task": null,
39
+ "forced_bos_token_id": null,
40
+ "forced_eos_token_id": null,
41
+ "hidden_act": "relu",
42
+ "hidden_sizes": [
43
+ 256,
44
+ 512,
45
+ 1024,
46
+ 2048
47
+ ],
48
+ "id2label": {
49
+ "0": "LABEL_0",
50
+ "1": "LABEL_1"
51
+ },
52
+ "is_decoder": false,
53
+ "is_encoder_decoder": false,
54
+ "label2id": {
55
+ "LABEL_0": 0,
56
+ "LABEL_1": 1
57
+ },
58
+ "layer_type": "bottleneck",
59
+ "length_penalty": 1.0,
60
+ "max_length": 20,
61
+ "min_length": 0,
62
+ "model_type": "resnet",
63
+ "no_repeat_ngram_size": 0,
64
+ "num_beam_groups": 1,
65
+ "num_beams": 1,
66
+ "num_channels": 3,
67
+ "num_return_sequences": 1,
68
+ "out_features": [
69
+ "stage4"
70
+ ],
71
+ "out_indices": [
72
+ 4
73
+ ],
74
+ "output_attentions": false,
75
+ "output_hidden_states": false,
76
+ "output_scores": false,
77
+ "pad_token_id": null,
78
+ "prefix": null,
79
+ "problem_type": null,
80
+ "pruned_heads": {},
81
+ "remove_invalid_values": false,
82
+ "repetition_penalty": 1.0,
83
+ "return_dict": true,
84
+ "return_dict_in_generate": false,
85
+ "sep_token_id": null,
86
+ "stage_names": [
87
+ "stem",
88
+ "stage1",
89
+ "stage2",
90
+ "stage3",
91
+ "stage4"
92
+ ],
93
+ "suppress_tokens": null,
94
+ "task_specific_params": null,
95
+ "temperature": 1.0,
96
+ "tf_legacy_loss": false,
97
+ "tie_encoder_decoder": false,
98
+ "tie_word_embeddings": true,
99
+ "tokenizer_class": null,
100
+ "top_k": 50,
101
+ "top_p": 1.0,
102
+ "torch_dtype": null,
103
+ "torchscript": false,
104
+ "typical_p": 1.0,
105
+ "use_bfloat16": false
106
+ },
107
+ "backbone_kwargs": null,
108
+ "bbox_cost": 5,
109
+ "bbox_loss_coefficient": 5,
110
+ "class_cost": 1,
111
+ "d_model": 256,
112
+ "decoder_attention_heads": 8,
113
+ "decoder_ffn_dim": 2048,
114
+ "decoder_layerdrop": 0.0,
115
+ "decoder_layers": 6,
116
+ "dice_loss_coefficient": 1,
117
+ "dilation": null,
118
+ "dropout": 0.1,
119
+ "encoder_attention_heads": 8,
120
+ "encoder_ffn_dim": 2048,
121
+ "encoder_layerdrop": 0.0,
122
+ "encoder_layers": 6,
123
+ "eos_coefficient": 0.1,
124
+ "giou_cost": 2,
125
+ "giou_loss_coefficient": 2,
126
+ "id2label": {
127
+ "0": "LABEL_0"
128
+ },
129
+ "init_std": 0.02,
130
+ "init_xavier_std": 1.0,
131
+ "is_encoder_decoder": true,
132
+ "label2id": {
133
+ "LABEL_0": 0
134
+ },
135
+ "mask_loss_coefficient": 1,
136
+ "model_type": "detr",
137
+ "num_channels": 3,
138
+ "num_hidden_layers": 6,
139
+ "num_queries": 100,
140
+ "position_embedding_type": "sine",
141
+ "scale_embedding": false,
142
+ "torch_dtype": "float32",
143
+ "transformers_version": "4.49.0",
144
+ "use_pretrained_backbone": null,
145
+ "use_timm_backbone": false
146
+ }
onnx/model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbfd223ec31755976b667407dbbb9befc8ac5e45a74c6852436e8e56c29e2783
3
+ size 166726523
onnx/model_bnb4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9e2e5738c4aa2a18267f4365d49ce07fc7871c8f9c3f8d217a56c9e4d6aabb7
3
+ size 106812274
onnx/model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5cb43c4e6231441b1bf85b012c5a61a49f7240778d81cb6133858478edc0b27
3
+ size 83620833
onnx/model_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3536c19006a0e48855af842cd0aa6d761d1cfe85ee9f876865775d1f15b52992
3
+ size 42965758
onnx/model_q4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b6383572dfd84b564b2356c5ad72fc8b27160ed9a2150aecf7455a7739e52248
3
+ size 107901132
onnx/model_q4f16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de5e79c81a05411afc47c9387901333b75ea29859036ad913e526d886b7126fa
3
+ size 58574080
onnx/model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50ba0451b54e5f1d926cbcb53710e2d1a64114be6a4846d173be76619c234468
3
+ size 42965758
onnx/model_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50ba0451b54e5f1d926cbcb53710e2d1a64114be6a4846d173be76619c234468
3
+ size 42965758
preprocessor_config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_annotations": true,
3
+ "do_normalize": true,
4
+ "do_pad": true,
5
+ "do_rescale": true,
6
+ "do_resize": true,
7
+ "format": "coco_detection",
8
+ "image_mean": [
9
+ 0.485,
10
+ 0.456,
11
+ 0.406
12
+ ],
13
+ "image_processor_type": "DetrFeatureExtractor",
14
+ "image_std": [
15
+ 0.229,
16
+ 0.224,
17
+ 0.225
18
+ ],
19
+ "pad_size": null,
20
+ "resample": 2,
21
+ "rescale_factor": 0.00392156862745098,
22
+ "size": {
23
+ "longest_edge": 1333,
24
+ "shortest_edge": 800
25
+ }
26
+ }
quantize_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "modes": [
3
+ "fp16",
4
+ "q8",
5
+ "int8",
6
+ "uint8",
7
+ "q4",
8
+ "q4f16",
9
+ "bnb4"
10
+ ],
11
+ "per_channel": true,
12
+ "reduce_range": true,
13
+ "block_size": null,
14
+ "is_symmetric": true,
15
+ "accuracy_level": null,
16
+ "quant_type": 1,
17
+ "op_block_list": null
18
+ }