Hugging Face's logo

keras
/
retinanet_resnet50_fpn_coco

KerasHub
Model card Files Community
Divyasreepat commited on
Commit
c866ac3
·
verified ·
1 Parent(s): 42d95fd

Upload folder using huggingface_hub

Browse files
Files changed (8) hide show
  1. README.md +17 -0
  2. config.json +58 -0
  3. image_converter.json +32 -0
  4. metadata.json +6 -0
  5. model.weights.h5 +3 -0
  6. preprocessor.json +50 -0
  7. task.json +225 -0
  8. task.weights.h5 +3 -0
README.md ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: keras-hub
3
+ ---
4
+ This is a [`RetinaNet` model](https://keras.io/api/keras_hub/models/retina_net) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
5
+ This model is related to a `ObjectDetector` task.
6
+
7
+ Model config:
8
+ * **name:** retina_net_backbone_1
9
+ * **trainable:** True
10
+ * **image_encoder:** {'module': 'keras_hub.src.models.resnet.resnet_backbone', 'class_name': 'ResNetBackbone', 'config': {'name': 'res_net_backbone', 'trainable': True, 'input_conv_filters': [64], 'input_conv_kernel_sizes': [7], 'stackwise_num_filters': [64, 128, 256, 512], 'stackwise_num_blocks': [3, 4, 6, 3], 'stackwise_num_strides': [1, 2, 2, 2], 'block_type': 'bottleneck_block', 'use_pre_activation': False, 'image_shape': [None, None, 3]}, 'registered_name': 'keras_hub>ResNetBackbone'}
11
+ * **min_level:** 3
12
+ * **max_level:** 7
13
+ * **use_p5:** False
14
+ * **use_fpn_batch_norm:** False
15
+ * **image_shape:** [None, None, 3]
16
+
17
+ This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.
config.json ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.retinanet.retinanet_backbone",
3
+ "class_name": "RetinaNetBackbone",
4
+ "config": {
5
+ "name": "retina_net_backbone_1",
6
+ "trainable": true,
7
+ "image_encoder": {
8
+ "module": "keras_hub.src.models.resnet.resnet_backbone",
9
+ "class_name": "ResNetBackbone",
10
+ "config": {
11
+ "name": "res_net_backbone",
12
+ "trainable": true,
13
+ "input_conv_filters": [
14
+ 64
15
+ ],
16
+ "input_conv_kernel_sizes": [
17
+ 7
18
+ ],
19
+ "stackwise_num_filters": [
20
+ 64,
21
+ 128,
22
+ 256,
23
+ 512
24
+ ],
25
+ "stackwise_num_blocks": [
26
+ 3,
27
+ 4,
28
+ 6,
29
+ 3
30
+ ],
31
+ "stackwise_num_strides": [
32
+ 1,
33
+ 2,
34
+ 2,
35
+ 2
36
+ ],
37
+ "block_type": "bottleneck_block",
38
+ "use_pre_activation": false,
39
+ "image_shape": [
40
+ null,
41
+ null,
42
+ 3
43
+ ]
44
+ },
45
+ "registered_name": "keras_hub>ResNetBackbone"
46
+ },
47
+ "min_level": 3,
48
+ "max_level": 7,
49
+ "use_p5": false,
50
+ "use_fpn_batch_norm": false,
51
+ "image_shape": [
52
+ null,
53
+ null,
54
+ 3
55
+ ]
56
+ },
57
+ "registered_name": "keras_hub>RetinaNetBackbone"
58
+ }
image_converter.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.retinanet.retinanet_image_converter",
3
+ "class_name": "RetinaNetImageConverter",
4
+ "config": {
5
+ "name": "retina_net_image_converter",
6
+ "trainable": true,
7
+ "dtype": {
8
+ "module": "keras",
9
+ "class_name": "DTypePolicy",
10
+ "config": {
11
+ "name": "float32"
12
+ },
13
+ "registered_name": null
14
+ },
15
+ "image_size": null,
16
+ "scale": 0.00392156862745098,
17
+ "offset": null,
18
+ "interpolation": "bilinear",
19
+ "crop_to_aspect_ratio": true,
20
+ "norm_mean": [
21
+ 0.485,
22
+ 0.456,
23
+ 0.406
24
+ ],
25
+ "norm_std": [
26
+ 0.229,
27
+ 0.224,
28
+ 0.225
29
+ ]
30
+ },
31
+ "registered_name": "keras_hub>RetinaNetImageConverter"
32
+ }
metadata.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "keras_version": "3.7.0",
3
+ "keras_hub_version": "0.17.0.dev0",
4
+ "parameter_count": 27429824,
5
+ "date_saved": "2024-11-07@21:59:35"
6
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:491b508fb613c8d0fd6c65875481a1c9932367b1280f02b7eda38666cac88ac9
3
+ size 110218880
preprocessor.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.retinanet.retinanet_object_detector_preprocessor",
3
+ "class_name": "RetinaNetObjectDetectorPreprocessor",
4
+ "config": {
5
+ "name": "retina_net_object_detector_preprocessor",
6
+ "trainable": true,
7
+ "dtype": {
8
+ "module": "keras",
9
+ "class_name": "DTypePolicy",
10
+ "config": {
11
+ "name": "float32"
12
+ },
13
+ "registered_name": null
14
+ },
15
+ "image_converter": {
16
+ "module": "keras_hub.src.models.retinanet.retinanet_image_converter",
17
+ "class_name": "RetinaNetImageConverter",
18
+ "config": {
19
+ "name": "retina_net_image_converter",
20
+ "trainable": true,
21
+ "dtype": {
22
+ "module": "keras",
23
+ "class_name": "DTypePolicy",
24
+ "config": {
25
+ "name": "float32"
26
+ },
27
+ "registered_name": null
28
+ },
29
+ "image_size": null,
30
+ "scale": 0.00392156862745098,
31
+ "offset": null,
32
+ "interpolation": "bilinear",
33
+ "crop_to_aspect_ratio": true,
34
+ "norm_mean": [
35
+ 0.485,
36
+ 0.456,
37
+ 0.406
38
+ ],
39
+ "norm_std": [
40
+ 0.229,
41
+ 0.224,
42
+ 0.225
43
+ ]
44
+ },
45
+ "registered_name": "keras_hub>RetinaNetImageConverter"
46
+ },
47
+ "config_file": "preprocessor.json"
48
+ },
49
+ "registered_name": "keras_hub>RetinaNetObjectDetectorPreprocessor"
50
+ }
task.json ADDED
@@ -0,0 +1,225 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "module": "keras_hub.src.models.retinanet.retinanet_object_detector",
3
+ "class_name": "RetinaNetObjectDetector",
4
+ "config": {
5
+ "backbone": {
6
+ "module": "keras_hub.src.models.retinanet.retinanet_backbone",
7
+ "class_name": "RetinaNetBackbone",
8
+ "config": {
9
+ "name": "retina_net_backbone_1",
10
+ "trainable": true,
11
+ "image_encoder": {
12
+ "module": "keras_hub.src.models.resnet.resnet_backbone",
13
+ "class_name": "ResNetBackbone",
14
+ "config": {
15
+ "name": "res_net_backbone",
16
+ "trainable": true,
17
+ "input_conv_filters": [
18
+ 64
19
+ ],
20
+ "input_conv_kernel_sizes": [
21
+ 7
22
+ ],
23
+ "stackwise_num_filters": [
24
+ 64,
25
+ 128,
26
+ 256,
27
+ 512
28
+ ],
29
+ "stackwise_num_blocks": [
30
+ 3,
31
+ 4,
32
+ 6,
33
+ 3
34
+ ],
35
+ "stackwise_num_strides": [
36
+ 1,
37
+ 2,
38
+ 2,
39
+ 2
40
+ ],
41
+ "block_type": "bottleneck_block",
42
+ "use_pre_activation": false,
43
+ "image_shape": [
44
+ null,
45
+ null,
46
+ 3
47
+ ]
48
+ },
49
+ "registered_name": "keras_hub>ResNetBackbone"
50
+ },
51
+ "min_level": 3,
52
+ "max_level": 7,
53
+ "use_p5": false,
54
+ "use_fpn_batch_norm": false,
55
+ "image_shape": [
56
+ null,
57
+ null,
58
+ 3
59
+ ]
60
+ },
61
+ "registered_name": "keras_hub>RetinaNetBackbone"
62
+ },
63
+ "preprocessor": {
64
+ "module": "keras_hub.src.models.retinanet.retinanet_object_detector_preprocessor",
65
+ "class_name": "RetinaNetObjectDetectorPreprocessor",
66
+ "config": {
67
+ "name": "retina_net_object_detector_preprocessor",
68
+ "trainable": true,
69
+ "dtype": {
70
+ "module": "keras",
71
+ "class_name": "DTypePolicy",
72
+ "config": {
73
+ "name": "float32"
74
+ },
75
+ "registered_name": null
76
+ },
77
+ "image_converter": {
78
+ "module": "keras_hub.src.models.retinanet.retinanet_image_converter",
79
+ "class_name": "RetinaNetImageConverter",
80
+ "config": {
81
+ "name": "retina_net_image_converter",
82
+ "trainable": true,
83
+ "dtype": {
84
+ "module": "keras",
85
+ "class_name": "DTypePolicy",
86
+ "config": {
87
+ "name": "float32"
88
+ },
89
+ "registered_name": null
90
+ },
91
+ "image_size": null,
92
+ "scale": 0.00392156862745098,
93
+ "offset": null,
94
+ "interpolation": "bilinear",
95
+ "crop_to_aspect_ratio": true,
96
+ "norm_mean": [
97
+ 0.485,
98
+ 0.456,
99
+ 0.406
100
+ ],
101
+ "norm_std": [
102
+ 0.229,
103
+ 0.224,
104
+ 0.225
105
+ ]
106
+ },
107
+ "registered_name": "keras_hub>RetinaNetImageConverter"
108
+ },
109
+ "config_file": "preprocessor.json"
110
+ },
111
+ "registered_name": "keras_hub>RetinaNetObjectDetectorPreprocessor"
112
+ },
113
+ "name": "retina_net_object_detector_1",
114
+ "num_classes": 91,
115
+ "use_prediction_head_norm": false,
116
+ "pre_logits_num_conv_layers": 4,
117
+ "bounding_box_format": "yxyx",
118
+ "anchor_generator": {
119
+ "module": "keras_hub.src.models.retinanet.anchor_generator",
120
+ "class_name": "AnchorGenerator",
121
+ "config": {
122
+ "name": "anchor_generator_1",
123
+ "min_level": 3,
124
+ "max_level": 7,
125
+ "num_scales": 3,
126
+ "aspect_ratios": [
127
+ 0.5,
128
+ 1.0,
129
+ 2.0
130
+ ],
131
+ "anchor_size": 4,
132
+ "bounding_box_format": "yxyx",
133
+ "trainable": true,
134
+ "dtype": {
135
+ "module": "keras",
136
+ "class_name": "DTypePolicy",
137
+ "config": {
138
+ "name": "float32"
139
+ },
140
+ "registered_name": null
141
+ }
142
+ },
143
+ "registered_name": "keras_hub>AnchorGenerator"
144
+ },
145
+ "label_encoder": {
146
+ "module": "keras_hub.src.models.retinanet.retinanet_label_encoder",
147
+ "class_name": "RetinaNetLabelEncoder",
148
+ "config": {
149
+ "name": "retina_net_label_encoder_1",
150
+ "trainable": true,
151
+ "dtype": {
152
+ "module": "keras",
153
+ "class_name": "DTypePolicy",
154
+ "config": {
155
+ "name": "float32"
156
+ },
157
+ "registered_name": null
158
+ },
159
+ "anchor_generator": {
160
+ "module": "keras_hub.src.models.retinanet.anchor_generator",
161
+ "class_name": "AnchorGenerator",
162
+ "config": {
163
+ "name": "anchor_generator_1",
164
+ "min_level": 3,
165
+ "max_level": 7,
166
+ "num_scales": 3,
167
+ "aspect_ratios": [
168
+ 0.5,
169
+ 1.0,
170
+ 2.0
171
+ ],
172
+ "anchor_size": 4,
173
+ "bounding_box_format": "yxyx",
174
+ "trainable": true,
175
+ "dtype": {
176
+ "module": "keras",
177
+ "class_name": "DTypePolicy",
178
+ "config": {
179
+ "name": "float32"
180
+ },
181
+ "registered_name": null
182
+ }
183
+ },
184
+ "registered_name": "keras_hub>AnchorGenerator"
185
+ },
186
+ "bounding_box_format": "yxyx",
187
+ "encoding_format": "center_xywh",
188
+ "positive_threshold": 0.5,
189
+ "box_variance": [
190
+ 1.0,
191
+ 1.0,
192
+ 1.0,
193
+ 1.0
194
+ ],
195
+ "negative_threshold": 0.4,
196
+ "background_class": -1.0,
197
+ "ignore_class": -2.0
198
+ },
199
+ "registered_name": "RetinaNetLabelEncoder"
200
+ },
201
+ "prediction_decoder": {
202
+ "module": "keras_hub.src.models.retinanet.non_max_supression",
203
+ "class_name": "NonMaxSuppression",
204
+ "config": {
205
+ "name": "non_max_suppression_1",
206
+ "trainable": true,
207
+ "dtype": {
208
+ "module": "keras",
209
+ "class_name": "DTypePolicy",
210
+ "config": {
211
+ "name": "float32"
212
+ },
213
+ "registered_name": null
214
+ },
215
+ "bounding_box_format": "yxyx",
216
+ "from_logits": true,
217
+ "iou_threshold": 0.5,
218
+ "confidence_threshold": 0.5,
219
+ "max_detections": 100
220
+ },
221
+ "registered_name": "NonMaxSuppression"
222
+ }
223
+ },
224
+ "registered_name": "keras_hub>RetinaNetObjectDetector"
225
+ }
task.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5639d31c84f8c282e6a06b8ee2d41ab4605d0ac50151d1130186b52b316897f1
3
+ size 137066612