22A223R commited on
Commit
d9b8e90
1 Parent(s): 7dd5fb5

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
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: 1704528026.470923
4
+ last_preserved_timestamp: 1704528025.3159719
checkpoint/ckpt-0.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fab1a09f1ffc665854890cf398da9dfa57739871af4259826f3d2ded2944f9f
3
+ size 18728120
checkpoint/ckpt-0.index ADDED
Binary file (22.2 kB). View file
 
pipeline.config ADDED
@@ -0,0 +1,188 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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/balloon_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
+ fine_tune_checkpoint_version: V2
170
+ }
171
+ train_input_reader {
172
+ label_map_path: "/home/ubuntu/balloon_project/data/label_map.pbtxt"
173
+ tf_record_input_reader {
174
+ input_path: "/home/ubuntu/balloon_project/data/train.record"
175
+ }
176
+ }
177
+ eval_config {
178
+ metrics_set: "coco_detection_metrics"
179
+ use_moving_averages: false
180
+ }
181
+ eval_input_reader {
182
+ label_map_path: "/home/ubuntu/balloon_project/data/label_map.pbtxt"
183
+ shuffle: false
184
+ num_epochs: 1
185
+ tf_record_input_reader {
186
+ input_path: "/home/ubuntu/balloon_project/data/val.record"
187
+ }
188
+ }
saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:045812faffae9b2cb683ad84045ed7e4073d50c19a32b742e65ce3ffbc31f781
3
+ size 8662866
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:8352b64d45ebd0299a33f2f0440334838cee5c08f7b1097cbaf269ee71c5aa79
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:55572b3dfe3a6dbe5bfd6a948408766e1efc40831b00e6d04b8968446b04d4fb
3
+ size 19057