wongshennan commited on
Commit
4595b4b
0 Parent(s):
.gitattributes ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz 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
README.md ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
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: 1705223532.7053046
4
+ last_preserved_timestamp: 1705223531.5241418
checkpoint/ckpt-0.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8cc95d37b40cedc92b677ae5c6057af84523ed4fa2371540842ee1820b5a244
3
+ size 10479755
checkpoint/ckpt-0.index ADDED
Binary file (26.2 kB). View file
 
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: 10
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: 25000
157
+ warmup_learning_rate: 0.026666
158
+ warmup_steps: 2000
159
+ }
160
+ }
161
+ momentum_optimizer_value: 0.9
162
+ }
163
+ use_moving_average: false
164
+ }
165
+ fine_tune_checkpoint: "/home/ubuntu/iti107-assignment2/pretrained_models/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: "/home/ubuntu/iti107-assignment2/data/label_map.pbtxt"
176
+ tf_record_input_reader {
177
+ input_path: "/home/ubuntu/iti107-assignment2/data/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: "/home/ubuntu/iti107-assignment2/data/label_map.pbtxt"
186
+ shuffle: false
187
+ num_epochs: 1
188
+ tf_record_input_reader {
189
+ input_path: "/home/ubuntu/iti107-assignment2/data/val.record"
190
+ }
191
+ }
saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e1a50c71f524048b86d81d36629417f0dd4254eecea91b351409b72e59af92a
3
+ size 9238055
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:705d80861f409f91a6142d8dbf285c44ab6710bfbe3b5535fd3a48d57fc36453
3
+ size 10549539
saved_model/variables/variables.index ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3312052241b9a88dd7036355ab0ae66d66f42047d14f1b06e482f04083c1e44
3
+ size 22366