alexrods commited on
Commit
bc6f589
1 Parent(s): 893799b

Upload model

Browse files
.gitattributes CHANGED
@@ -32,3 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
32
  *.zip filter=lfs diff=lfs merge=lfs -text
33
  *.zst filter=lfs diff=lfs merge=lfs -text
34
  *tfevents* filter=lfs diff=lfs merge=lfs -text
35
+ model/checkpoint/ckpt-0.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
36
+ model/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
model/checkpoint/checkpoint ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ model_checkpoint_path: "ckpt-0"
2
+ all_model_checkpoint_paths: "ckpt-0"
3
+ all_model_checkpoint_timestamps: 1678645136.3784287
4
+ last_preserved_timestamp: 1678645135.1743114
model/checkpoint/ckpt-0.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25842c1b27201517de72d4365556b13601ffccb4d84c5b4abda9bb3381c96589
3
+ size 10479785
model/checkpoint/ckpt-0.index ADDED
Binary file (26.2 kB). View file
 
model/label_map.pbtxt ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ item {
2
+ name:'carro'
3
+ id:1
4
+ }
5
+ item {
6
+ name:'motos'
7
+ id:2
8
+ }
model/pipeline.config ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ ssd {
3
+ num_classes: 2
4
+ image_resizer {
5
+ fixed_shape_resizer {
6
+ height: 320
7
+ width: 320
8
+ }
9
+ }
10
+ feature_extractor {
11
+ type: "ssd_mobilenet_v2_fpn_keras"
12
+ depth_multiplier: 1.0
13
+ min_depth: 16
14
+ conv_hyperparams {
15
+ regularizer {
16
+ l2_regularizer {
17
+ weight: 4e-05
18
+ }
19
+ }
20
+ initializer {
21
+ random_normal_initializer {
22
+ mean: 0.0
23
+ stddev: 0.01
24
+ }
25
+ }
26
+ activation: RELU_6
27
+ batch_norm {
28
+ decay: 0.997
29
+ scale: true
30
+ epsilon: 0.001
31
+ }
32
+ }
33
+ use_depthwise: true
34
+ override_base_feature_extractor_hyperparams: true
35
+ fpn {
36
+ min_level: 3
37
+ max_level: 7
38
+ additional_layer_depth: 128
39
+ }
40
+ }
41
+ box_coder {
42
+ faster_rcnn_box_coder {
43
+ y_scale: 10.0
44
+ x_scale: 10.0
45
+ height_scale: 5.0
46
+ width_scale: 5.0
47
+ }
48
+ }
49
+ matcher {
50
+ argmax_matcher {
51
+ matched_threshold: 0.5
52
+ unmatched_threshold: 0.5
53
+ ignore_thresholds: false
54
+ negatives_lower_than_unmatched: true
55
+ force_match_for_each_row: true
56
+ use_matmul_gather: true
57
+ }
58
+ }
59
+ similarity_calculator {
60
+ iou_similarity {
61
+ }
62
+ }
63
+ box_predictor {
64
+ weight_shared_convolutional_box_predictor {
65
+ conv_hyperparams {
66
+ regularizer {
67
+ l2_regularizer {
68
+ weight: 4e-05
69
+ }
70
+ }
71
+ initializer {
72
+ random_normal_initializer {
73
+ mean: 0.0
74
+ stddev: 0.01
75
+ }
76
+ }
77
+ activation: RELU_6
78
+ batch_norm {
79
+ decay: 0.997
80
+ scale: true
81
+ epsilon: 0.001
82
+ }
83
+ }
84
+ depth: 128
85
+ num_layers_before_predictor: 4
86
+ kernel_size: 3
87
+ class_prediction_bias_init: -4.6
88
+ share_prediction_tower: true
89
+ use_depthwise: true
90
+ }
91
+ }
92
+ anchor_generator {
93
+ multiscale_anchor_generator {
94
+ min_level: 3
95
+ max_level: 7
96
+ anchor_scale: 4.0
97
+ aspect_ratios: 1.0
98
+ aspect_ratios: 2.0
99
+ aspect_ratios: 0.5
100
+ scales_per_octave: 2
101
+ }
102
+ }
103
+ post_processing {
104
+ batch_non_max_suppression {
105
+ score_threshold: 1e-08
106
+ iou_threshold: 0.6
107
+ max_detections_per_class: 100
108
+ max_total_detections: 100
109
+ use_static_shapes: false
110
+ }
111
+ score_converter: SIGMOID
112
+ }
113
+ normalize_loss_by_num_matches: true
114
+ loss {
115
+ localization_loss {
116
+ weighted_smooth_l1 {
117
+ }
118
+ }
119
+ classification_loss {
120
+ weighted_sigmoid_focal {
121
+ gamma: 2.0
122
+ alpha: 0.25
123
+ }
124
+ }
125
+ classification_weight: 1.0
126
+ localization_weight: 1.0
127
+ }
128
+ encode_background_as_zeros: true
129
+ normalize_loc_loss_by_codesize: true
130
+ inplace_batchnorm_update: true
131
+ freeze_batchnorm: false
132
+ }
133
+ }
134
+ train_config {
135
+ batch_size: 4
136
+ data_augmentation_options {
137
+ random_horizontal_flip {
138
+ }
139
+ }
140
+ data_augmentation_options {
141
+ random_crop_image {
142
+ min_object_covered: 0.0
143
+ min_aspect_ratio: 0.75
144
+ max_aspect_ratio: 3.0
145
+ min_area: 0.75
146
+ max_area: 1.0
147
+ overlap_thresh: 0.0
148
+ }
149
+ }
150
+ sync_replicas: true
151
+ optimizer {
152
+ momentum_optimizer {
153
+ learning_rate {
154
+ cosine_decay_learning_rate {
155
+ learning_rate_base: 0.08
156
+ total_steps: 50000
157
+ warmup_learning_rate: 0.026666
158
+ warmup_steps: 1000
159
+ }
160
+ }
161
+ momentum_optimizer_value: 0.9
162
+ }
163
+ use_moving_average: false
164
+ }
165
+ fine_tune_checkpoint: "/content/models/research/ssd_mobilenet_v2_fpnlite_320x320_coco17_tpu-8/checkpoint/ckpt-0"
166
+ num_steps: 50000
167
+ startup_delay_steps: 0.0
168
+ replicas_to_aggregate: 8
169
+ max_number_of_boxes: 100
170
+ unpad_groundtruth_tensors: false
171
+ fine_tune_checkpoint_type: "detection"
172
+ fine_tune_checkpoint_version: V2
173
+ }
174
+ train_input_reader {
175
+ label_map_path: "/content/label_map.pbtxt"
176
+ tf_record_input_reader {
177
+ input_path: "/content/train.record"
178
+ }
179
+ }
180
+ eval_config {
181
+ metrics_set: "coco_detection_metrics"
182
+ use_moving_averages: false
183
+ }
184
+ eval_input_reader {
185
+ label_map_path: "/content/label_map.pbtxt"
186
+ shuffle: false
187
+ num_epochs: 1
188
+ tf_record_input_reader {
189
+ input_path: "/content/test.record"
190
+ }
191
+ }
model/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:723664b8a304215599d74e06d6344d136cd814afa1792820b371b63f50897fd1
3
+ size 7711722
model/saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc910673d396503b755a98c008d0bdeaad53cc6d4bc20fbd3c33b03b88beacf3
3
+ size 10547253
model/saved_model/variables/variables.index ADDED
Binary file (22.1 kB). View file