Wang commited on
Commit
83e809f
1 Parent(s): bcf7d4a

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ checkpoint/ckpt-0.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
.ipynb_checkpoints/pipeline-checkpoint.config ADDED
@@ -0,0 +1,189 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ ssd {
3
+ num_classes: 2
4
+ image_resizer {
5
+ keep_aspect_ratio_resizer {
6
+ min_dimension: 512
7
+ max_dimension: 512
8
+ pad_to_max_dimension: true
9
+ }
10
+ }
11
+ feature_extractor {
12
+ type: "ssd_mobilenet_v2_keras"
13
+ depth_multiplier: 1.0
14
+ min_depth: 16
15
+ conv_hyperparams {
16
+ regularizer {
17
+ l2_regularizer {
18
+ weight: 4e-05
19
+ }
20
+ }
21
+ initializer {
22
+ truncated_normal_initializer {
23
+ mean: 0.0
24
+ stddev: 0.03
25
+ }
26
+ }
27
+ activation: RELU_6
28
+ batch_norm {
29
+ decay: 0.97
30
+ center: true
31
+ scale: true
32
+ epsilon: 0.001
33
+ train: true
34
+ }
35
+ }
36
+ override_base_feature_extractor_hyperparams: true
37
+ }
38
+ box_coder {
39
+ faster_rcnn_box_coder {
40
+ y_scale: 10.0
41
+ x_scale: 10.0
42
+ height_scale: 5.0
43
+ width_scale: 5.0
44
+ }
45
+ }
46
+ matcher {
47
+ argmax_matcher {
48
+ matched_threshold: 0.5
49
+ unmatched_threshold: 0.5
50
+ ignore_thresholds: false
51
+ negatives_lower_than_unmatched: true
52
+ force_match_for_each_row: true
53
+ use_matmul_gather: true
54
+ }
55
+ }
56
+ similarity_calculator {
57
+ iou_similarity {
58
+ }
59
+ }
60
+ box_predictor {
61
+ convolutional_box_predictor {
62
+ conv_hyperparams {
63
+ regularizer {
64
+ l2_regularizer {
65
+ weight: 4e-05
66
+ }
67
+ }
68
+ initializer {
69
+ random_normal_initializer {
70
+ mean: 0.0
71
+ stddev: 0.01
72
+ }
73
+ }
74
+ activation: RELU_6
75
+ batch_norm {
76
+ decay: 0.97
77
+ center: true
78
+ scale: true
79
+ epsilon: 0.001
80
+ train: true
81
+ }
82
+ }
83
+ min_depth: 0
84
+ max_depth: 0
85
+ num_layers_before_predictor: 0
86
+ use_dropout: false
87
+ dropout_keep_probability: 0.8
88
+ kernel_size: 1
89
+ box_code_size: 4
90
+ apply_sigmoid_to_scores: false
91
+ class_prediction_bias_init: -4.6
92
+ }
93
+ }
94
+ anchor_generator {
95
+ ssd_anchor_generator {
96
+ num_layers: 6
97
+ min_scale: 0.2
98
+ max_scale: 0.95
99
+ aspect_ratios: 1.0
100
+ aspect_ratios: 2.0
101
+ aspect_ratios: 0.5
102
+ aspect_ratios: 3.0
103
+ aspect_ratios: 0.3333
104
+ }
105
+ }
106
+ post_processing {
107
+ batch_non_max_suppression {
108
+ score_threshold: 1e-08
109
+ iou_threshold: 0.6
110
+ max_detections_per_class: 100
111
+ max_total_detections: 100
112
+ use_static_shapes: false
113
+ }
114
+ score_converter: SIGMOID
115
+ }
116
+ normalize_loss_by_num_matches: true
117
+ loss {
118
+ localization_loss {
119
+ weighted_smooth_l1 {
120
+ delta: 1.0
121
+ }
122
+ }
123
+ classification_loss {
124
+ weighted_sigmoid_focal {
125
+ gamma: 2.0
126
+ alpha: 0.75
127
+ }
128
+ }
129
+ classification_weight: 1.0
130
+ localization_weight: 1.0
131
+ }
132
+ encode_background_as_zeros: true
133
+ normalize_loc_loss_by_codesize: true
134
+ inplace_batchnorm_update: true
135
+ freeze_batchnorm: false
136
+ }
137
+ }
138
+ train_config {
139
+ batch_size: 10
140
+ data_augmentation_options {
141
+ random_horizontal_flip {
142
+ }
143
+ }
144
+ data_augmentation_options {
145
+ ssd_random_crop {
146
+ }
147
+ }
148
+ sync_replicas: true
149
+ optimizer {
150
+ momentum_optimizer {
151
+ learning_rate {
152
+ cosine_decay_learning_rate {
153
+ learning_rate_base: 0.04
154
+ total_steps: 25000
155
+ warmup_learning_rate: 0.013333
156
+ warmup_steps: 2000
157
+ }
158
+ }
159
+ momentum_optimizer_value: 0.9
160
+ }
161
+ use_moving_average: false
162
+ }
163
+ fine_tune_checkpoint: "/home/ubuntu/fastfood_project/pretrained_models/ssd_mobilenet_v2_320x320_coco17_tpu-8/checkpoint/ckpt-0"
164
+ num_steps: 50000
165
+ startup_delay_steps: 0.0
166
+ replicas_to_aggregate: 8
167
+ max_number_of_boxes: 100
168
+ unpad_groundtruth_tensors: false
169
+ fine_tune_checkpoint_type: "detection"
170
+ fine_tune_checkpoint_version: V2
171
+ }
172
+ train_input_reader {
173
+ label_map_path: "/home/ubuntu/fastfood_project/data/label_map.pbtxt"
174
+ tf_record_input_reader {
175
+ input_path: "/home/ubuntu/fastfood_project/data/train.record"
176
+ }
177
+ }
178
+ eval_config {
179
+ metrics_set: "coco_detection_metrics"
180
+ use_moving_averages: false
181
+ }
182
+ eval_input_reader {
183
+ label_map_path: "/home/ubuntu/fastfood_project/data/label_map.pbtxt"
184
+ shuffle: false
185
+ num_epochs: 1
186
+ tf_record_input_reader {
187
+ input_path: "/home/ubuntu/fastfood_project/data/val.record"
188
+ }
189
+ }
checkpoint/.ipynb_checkpoints/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: 1704271961.8282745
4
+ last_preserved_timestamp: 1704271960.6730993
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: 1704271961.8282745
4
+ last_preserved_timestamp: 1704271960.6730993
checkpoint/ckpt-0.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0648e78b6b81a77b35df03d14222823b2254e86729ca0e5bd0631d1a4ec85204
3
+ size 18728120
checkpoint/ckpt-0.index ADDED
Binary file (22.2 kB). View file
 
pipeline.config ADDED
@@ -0,0 +1,189 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model {
2
+ ssd {
3
+ num_classes: 2
4
+ image_resizer {
5
+ keep_aspect_ratio_resizer {
6
+ min_dimension: 512
7
+ max_dimension: 512
8
+ pad_to_max_dimension: true
9
+ }
10
+ }
11
+ feature_extractor {
12
+ type: "ssd_mobilenet_v2_keras"
13
+ depth_multiplier: 1.0
14
+ min_depth: 16
15
+ conv_hyperparams {
16
+ regularizer {
17
+ l2_regularizer {
18
+ weight: 4e-05
19
+ }
20
+ }
21
+ initializer {
22
+ truncated_normal_initializer {
23
+ mean: 0.0
24
+ stddev: 0.03
25
+ }
26
+ }
27
+ activation: RELU_6
28
+ batch_norm {
29
+ decay: 0.97
30
+ center: true
31
+ scale: true
32
+ epsilon: 0.001
33
+ train: true
34
+ }
35
+ }
36
+ override_base_feature_extractor_hyperparams: true
37
+ }
38
+ box_coder {
39
+ faster_rcnn_box_coder {
40
+ y_scale: 10.0
41
+ x_scale: 10.0
42
+ height_scale: 5.0
43
+ width_scale: 5.0
44
+ }
45
+ }
46
+ matcher {
47
+ argmax_matcher {
48
+ matched_threshold: 0.5
49
+ unmatched_threshold: 0.5
50
+ ignore_thresholds: false
51
+ negatives_lower_than_unmatched: true
52
+ force_match_for_each_row: true
53
+ use_matmul_gather: true
54
+ }
55
+ }
56
+ similarity_calculator {
57
+ iou_similarity {
58
+ }
59
+ }
60
+ box_predictor {
61
+ convolutional_box_predictor {
62
+ conv_hyperparams {
63
+ regularizer {
64
+ l2_regularizer {
65
+ weight: 4e-05
66
+ }
67
+ }
68
+ initializer {
69
+ random_normal_initializer {
70
+ mean: 0.0
71
+ stddev: 0.01
72
+ }
73
+ }
74
+ activation: RELU_6
75
+ batch_norm {
76
+ decay: 0.97
77
+ center: true
78
+ scale: true
79
+ epsilon: 0.001
80
+ train: true
81
+ }
82
+ }
83
+ min_depth: 0
84
+ max_depth: 0
85
+ num_layers_before_predictor: 0
86
+ use_dropout: false
87
+ dropout_keep_probability: 0.8
88
+ kernel_size: 1
89
+ box_code_size: 4
90
+ apply_sigmoid_to_scores: false
91
+ class_prediction_bias_init: -4.6
92
+ }
93
+ }
94
+ anchor_generator {
95
+ ssd_anchor_generator {
96
+ num_layers: 6
97
+ min_scale: 0.2
98
+ max_scale: 0.95
99
+ aspect_ratios: 1.0
100
+ aspect_ratios: 2.0
101
+ aspect_ratios: 0.5
102
+ aspect_ratios: 3.0
103
+ aspect_ratios: 0.3333
104
+ }
105
+ }
106
+ post_processing {
107
+ batch_non_max_suppression {
108
+ score_threshold: 1e-08
109
+ iou_threshold: 0.6
110
+ max_detections_per_class: 100
111
+ max_total_detections: 100
112
+ use_static_shapes: false
113
+ }
114
+ score_converter: SIGMOID
115
+ }
116
+ normalize_loss_by_num_matches: true
117
+ loss {
118
+ localization_loss {
119
+ weighted_smooth_l1 {
120
+ delta: 1.0
121
+ }
122
+ }
123
+ classification_loss {
124
+ weighted_sigmoid_focal {
125
+ gamma: 2.0
126
+ alpha: 0.75
127
+ }
128
+ }
129
+ classification_weight: 1.0
130
+ localization_weight: 1.0
131
+ }
132
+ encode_background_as_zeros: true
133
+ normalize_loc_loss_by_codesize: true
134
+ inplace_batchnorm_update: true
135
+ freeze_batchnorm: false
136
+ }
137
+ }
138
+ train_config {
139
+ batch_size: 10
140
+ data_augmentation_options {
141
+ random_horizontal_flip {
142
+ }
143
+ }
144
+ data_augmentation_options {
145
+ ssd_random_crop {
146
+ }
147
+ }
148
+ sync_replicas: true
149
+ optimizer {
150
+ momentum_optimizer {
151
+ learning_rate {
152
+ cosine_decay_learning_rate {
153
+ learning_rate_base: 0.04
154
+ total_steps: 25000
155
+ warmup_learning_rate: 0.013333
156
+ warmup_steps: 2000
157
+ }
158
+ }
159
+ momentum_optimizer_value: 0.9
160
+ }
161
+ use_moving_average: false
162
+ }
163
+ fine_tune_checkpoint: "/home/ubuntu/fastfood_project/pretrained_models/ssd_mobilenet_v2_320x320_coco17_tpu-8/checkpoint/ckpt-0"
164
+ num_steps: 50000
165
+ startup_delay_steps: 0.0
166
+ replicas_to_aggregate: 8
167
+ max_number_of_boxes: 100
168
+ unpad_groundtruth_tensors: false
169
+ fine_tune_checkpoint_type: "detection"
170
+ fine_tune_checkpoint_version: V2
171
+ }
172
+ train_input_reader {
173
+ label_map_path: "/home/ubuntu/fastfood_project/data/label_map.pbtxt"
174
+ tf_record_input_reader {
175
+ input_path: "/home/ubuntu/fastfood_project/data/train.record"
176
+ }
177
+ }
178
+ eval_config {
179
+ metrics_set: "coco_detection_metrics"
180
+ use_moving_averages: false
181
+ }
182
+ eval_input_reader {
183
+ label_map_path: "/home/ubuntu/fastfood_project/data/label_map.pbtxt"
184
+ shuffle: false
185
+ num_epochs: 1
186
+ tf_record_input_reader {
187
+ input_path: "/home/ubuntu/fastfood_project/data/val.record"
188
+ }
189
+ }
saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:852a5e9fb245d4c41b1cf7d5c525cdecaf838ba49565b4e60818f9f8ace3df15
3
+ size 8728182
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:500461663f1880aed84f1bf1a05b8a70f65d49132766382d09fd6689b0507669
3
+ size 18792474
saved_model/variables/variables.index ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b606d9ebe9e6e60478f3d096d614205769edf69b67c61d78813f6cb5e243a552
3
+ size 19057