Showing
21 changed files
with
237 additions
and
0 deletions
model_ckpt/try_2020_04_27/checkpoint
0 → 100644
No preview for this file type
model_ckpt/try_2020_04_27/events.out.tfevents.1587961082.cmle-training-6397358976514952948
0 → 100644
This file is too large to display.
model_ckpt/try_2020_04_27/graph.pbtxt
0 → 100644
This diff could not be displayed because it is too large.
model_ckpt/try_2020_04_27/imgs/mAP@50IOU.JPG
0 → 100644
32.7 KB
model_ckpt/try_2020_04_27/imgs/result1.JPG
0 → 100644
229 KB
model_ckpt/try_2020_04_27/imgs/result2.JPG
0 → 100644
152 KB
This file is too large to display.
model_ckpt/try_2020_04_27/model.ckpt-0.index
0 → 100644
No preview for this file type
model_ckpt/try_2020_04_27/model.ckpt-0.meta
0 → 100644
No preview for this file type
This file is too large to display.
No preview for this file type
No preview for this file type
This file is too large to display.
No preview for this file type
No preview for this file type
No preview for this file type
No preview for this file type
No preview for this file type
model_ckpt/try_2020_04_27/pipeline.config
0 → 100644
1 | +model { | ||
2 | + ssd { | ||
3 | + num_classes: 1 | ||
4 | + image_resizer { | ||
5 | + fixed_shape_resizer { | ||
6 | + height: 300 | ||
7 | + width: 300 | ||
8 | + } | ||
9 | + } | ||
10 | + feature_extractor { | ||
11 | + type: "ssd_mobilenet_v1_ppn" | ||
12 | + conv_hyperparams { | ||
13 | + regularizer { | ||
14 | + l2_regularizer { | ||
15 | + weight: 3.9999999e-05 | ||
16 | + } | ||
17 | + } | ||
18 | + initializer { | ||
19 | + random_normal_initializer { | ||
20 | + mean: 0.0 | ||
21 | + stddev: 0.0099999998 | ||
22 | + } | ||
23 | + } | ||
24 | + activation: RELU_6 | ||
25 | + batch_norm { | ||
26 | + decay: 0.97000003 | ||
27 | + center: true | ||
28 | + scale: true | ||
29 | + epsilon: 0.001 | ||
30 | + } | ||
31 | + } | ||
32 | + override_base_feature_extractor_hyperparams: true | ||
33 | + } | ||
34 | + box_coder { | ||
35 | + faster_rcnn_box_coder { | ||
36 | + y_scale: 10.0 | ||
37 | + x_scale: 10.0 | ||
38 | + height_scale: 5.0 | ||
39 | + width_scale: 5.0 | ||
40 | + } | ||
41 | + } | ||
42 | + matcher { | ||
43 | + argmax_matcher { | ||
44 | + matched_threshold: 0.5 | ||
45 | + unmatched_threshold: 0.5 | ||
46 | + ignore_thresholds: false | ||
47 | + negatives_lower_than_unmatched: true | ||
48 | + force_match_for_each_row: true | ||
49 | + use_matmul_gather: true | ||
50 | + } | ||
51 | + } | ||
52 | + similarity_calculator { | ||
53 | + iou_similarity { | ||
54 | + } | ||
55 | + } | ||
56 | + box_predictor { | ||
57 | + weight_shared_convolutional_box_predictor { | ||
58 | + conv_hyperparams { | ||
59 | + regularizer { | ||
60 | + l2_regularizer { | ||
61 | + weight: 3.9999999e-05 | ||
62 | + } | ||
63 | + } | ||
64 | + initializer { | ||
65 | + random_normal_initializer { | ||
66 | + mean: 0.0 | ||
67 | + stddev: 0.0099999998 | ||
68 | + } | ||
69 | + } | ||
70 | + activation: RELU_6 | ||
71 | + batch_norm { | ||
72 | + decay: 0.97000003 | ||
73 | + center: true | ||
74 | + scale: true | ||
75 | + epsilon: 0.001 | ||
76 | + train: true | ||
77 | + } | ||
78 | + } | ||
79 | + depth: 512 | ||
80 | + num_layers_before_predictor: 1 | ||
81 | + kernel_size: 1 | ||
82 | + class_prediction_bias_init: -4.5999999 | ||
83 | + share_prediction_tower: true | ||
84 | + } | ||
85 | + } | ||
86 | + anchor_generator { | ||
87 | + ssd_anchor_generator { | ||
88 | + num_layers: 6 | ||
89 | + min_scale: 0.15000001 | ||
90 | + max_scale: 0.94999999 | ||
91 | + aspect_ratios: 1.0 | ||
92 | + aspect_ratios: 2.0 | ||
93 | + aspect_ratios: 0.5 | ||
94 | + aspect_ratios: 3.0 | ||
95 | + aspect_ratios: 0.33329999 | ||
96 | + reduce_boxes_in_lowest_layer: false | ||
97 | + } | ||
98 | + } | ||
99 | + post_processing { | ||
100 | + batch_non_max_suppression { | ||
101 | + score_threshold: 9.9999999e-09 | ||
102 | + iou_threshold: 0.60000002 | ||
103 | + max_detections_per_class: 100 | ||
104 | + max_total_detections: 100 | ||
105 | + } | ||
106 | + score_converter: SIGMOID | ||
107 | + } | ||
108 | + normalize_loss_by_num_matches: true | ||
109 | + loss { | ||
110 | + localization_loss { | ||
111 | + weighted_smooth_l1 { | ||
112 | + } | ||
113 | + } | ||
114 | + classification_loss { | ||
115 | + weighted_sigmoid_focal { | ||
116 | + gamma: 2.0 | ||
117 | + alpha: 0.75 | ||
118 | + } | ||
119 | + } | ||
120 | + classification_weight: 1.0 | ||
121 | + localization_weight: 1.5 | ||
122 | + } | ||
123 | + encode_background_as_zeros: true | ||
124 | + normalize_loc_loss_by_codesize: true | ||
125 | + inplace_batchnorm_update: true | ||
126 | + freeze_batchnorm: false | ||
127 | + } | ||
128 | +} | ||
129 | +train_config { | ||
130 | + batch_size: 32 | ||
131 | + data_augmentation_options { | ||
132 | + random_horizontal_flip { | ||
133 | + } | ||
134 | + } | ||
135 | + data_augmentation_options { | ||
136 | + ssd_random_crop { | ||
137 | + } | ||
138 | + } | ||
139 | + sync_replicas: true | ||
140 | + optimizer { | ||
141 | + momentum_optimizer { | ||
142 | + learning_rate { | ||
143 | + cosine_decay_learning_rate { | ||
144 | + learning_rate_base: 0.69999999 | ||
145 | + total_steps: 50000 | ||
146 | + warmup_learning_rate: 0.13330001 | ||
147 | + warmup_steps: 2000 | ||
148 | + } | ||
149 | + } | ||
150 | + momentum_optimizer_value: 0.89999998 | ||
151 | + } | ||
152 | + use_moving_average: false | ||
153 | + } | ||
154 | + fine_tune_checkpoint: "gs://fire_land/data/model.ckpt" | ||
155 | + num_steps: 50000 | ||
156 | + startup_delay_steps: 0.0 | ||
157 | + replicas_to_aggregate: 8 | ||
158 | + max_number_of_boxes: 100 | ||
159 | + unpad_groundtruth_tensors: false | ||
160 | + use_bfloat16: false | ||
161 | +} | ||
162 | +train_input_reader { | ||
163 | + label_map_path: "gs://fire_land/data/fire_label_map.pbtxt" | ||
164 | + tf_record_input_reader { | ||
165 | + input_path: "gs://fire_land/data/train.record" | ||
166 | + } | ||
167 | +} | ||
168 | +eval_config { | ||
169 | + num_examples: 65 | ||
170 | + metrics_set: "coco_detection_metrics" | ||
171 | + use_moving_averages: false | ||
172 | +} | ||
173 | +eval_input_reader { | ||
174 | + label_map_path: "gs://fire_land/data/fire_label_map.pbtxt" | ||
175 | + shuffle: false | ||
176 | + num_epochs: 1 | ||
177 | + num_readers: 1 | ||
178 | + tf_record_input_reader { | ||
179 | + input_path: "gs://fire_land/data/val.record" | ||
180 | + } | ||
181 | + sample_1_of_n_examples: 1 | ||
182 | +} |
model_ckpt/try_2020_04_27/training_report.md
0 → 100644
1 | +# Training Report | ||
2 | + | ||
3 | +date: 2020 04 27 Mon | ||
4 | + | ||
5 | +model name: MobileNet based Single Shot MultiBox Detector with Pooling Pyramid Network | ||
6 | + | ||
7 | +input image size: 300 x 300 | ||
8 | + | ||
9 | +entire steps: 50000 | ||
10 | + | ||
11 | +best step: 32700 | ||
12 | + | ||
13 | +batch size: 32 | ||
14 | + | ||
15 | +number of samples for training: 300 | ||
16 | + | ||
17 | +number of samples for validation: 65 | ||
18 | + | ||
19 | +metrics set: coco detection metrics | ||
20 | + | ||
21 | + | ||
22 | + | ||
23 | +### mAP@50IOU | ||
24 | + | ||
25 | + | ||
26 | + | ||
27 | +**[32700 - best step]** | ||
28 | + | ||
29 | +mAP@50IOU: 64% | ||
30 | + | ||
31 | +loss: | ||
32 | + | ||
33 | + classification: 0.7869 | ||
34 | + | ||
35 | + localization: 0.4739 | ||
36 | + | ||
37 | + total: 1.418 | ||
38 | + | ||
39 | + | ||
40 | + | ||
41 | +### Result | ||
42 | + | ||
43 | + | ||
44 | + | ||
45 | + | ||
46 | + | ||
47 | + | ||
48 | + | ||
49 | +### Comment | ||
50 | + | ||
51 | +need more fire data. | ||
... | \ No newline at end of file | ... | \ No newline at end of file |
-
Please register or login to post a comment