23b719w commited on
Commit
d237924
1 Parent(s): cb44500

Upload 7 files

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