Advertisement
Guest User

Untitled

a guest
May 18th, 2019
327
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 4.62 KB | None | 0 0
  1. model {
  2. ssd {
  3. num_classes: 1
  4. image_resizer {
  5. fixed_shape_resizer {
  6. height: 256
  7. width: 256
  8. }
  9. }
  10. feature_extractor {
  11. type: "ssd_mobilenet_v1"
  12. depth_multiplier: 0.125
  13. min_depth: 16
  14. conv_hyperparams {
  15. regularizer {
  16. l2_regularizer {
  17. weight: 3.9999998989515007e-05
  18. }
  19. }
  20. initializer {
  21. truncated_normal_initializer {
  22. mean: 0.0
  23. stddev: 0.029999999329447746
  24. }
  25. }
  26. activation: RELU_6
  27. batch_norm {
  28. decay: 0.9700000286102295
  29. center: true
  30. scale: true
  31. epsilon: 0.0010000000474974513
  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: 3.9999998989515007e-05
  65. }
  66. }
  67. initializer {
  68. random_normal_initializer {
  69. mean: 0.0
  70. stddev: 0.009999999776482582
  71. }
  72. }
  73. activation: RELU_6
  74. batch_norm {
  75. decay: 0.9700000286102295
  76. center: true
  77. scale: true
  78. epsilon: 0.0010000000474974513
  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.800000011920929
  87. kernel_size: 1
  88. box_code_size: 4
  89. apply_sigmoid_to_scores: false
  90. class_prediction_bias_init: -4.599999904632568
  91. }
  92. }
  93. anchor_generator {
  94. ssd_anchor_generator {
  95. num_layers: 6
  96. min_scale: 0.20000000298023224
  97. max_scale: 0.949999988079071
  98. aspect_ratios: 1.0
  99. aspect_ratios: 2.0
  100. aspect_ratios: 0.5
  101. aspect_ratios: 3.0
  102. aspect_ratios: 0.33329999446868896
  103. }
  104. }
  105. post_processing {
  106. batch_non_max_suppression {
  107. score_threshold: 0.30000001192092896
  108. iou_threshold: 0.6000000238418579
  109. max_detections_per_class: 100
  110. max_total_detections: 100
  111. }
  112. score_converter: SIGMOID
  113. }
  114. normalize_loss_by_num_matches: true
  115. loss {
  116. localization_loss {
  117. weighted_smooth_l1 {
  118. delta: 1.0
  119. }
  120. }
  121. classification_loss {
  122. weighted_sigmoid_focal {
  123. gamma: 2.0
  124. alpha: 0.75
  125. }
  126. }
  127. classification_weight: 1.0
  128. localization_weight: 1.0
  129. }
  130. encode_background_as_zeros: true
  131. normalize_loc_loss_by_codesize: true
  132. inplace_batchnorm_update: true
  133. freeze_batchnorm: false
  134. }
  135. }
  136. train_config {
  137. batch_size: 384
  138. data_augmentation_options {
  139. random_horizontal_flip {
  140. }
  141. }
  142. data_augmentation_options {
  143. ssd_random_crop {
  144. }
  145. }
  146. sync_replicas: true
  147. optimizer {
  148. momentum_optimizer {
  149. learning_rate {
  150. cosine_decay_learning_rate {
  151. learning_rate_base: 0.8999999761581421
  152. total_steps: 10000
  153. warmup_learning_rate: 0.30000001192092896
  154. warmup_steps: 300
  155. }
  156. }
  157. momentum_optimizer_value: 0.8999999761581421
  158. }
  159. use_moving_average: false
  160. }
  161. #fine_tune_checkpoint: "gs://tfstorage2/data/model.ckpt"
  162. num_steps: 200000
  163. startup_delay_steps: 0.0
  164. replicas_to_aggregate: 8
  165. max_number_of_boxes: 100
  166. unpad_groundtruth_tensors: false
  167. fine_tune_checkpoint_type: "detection"
  168. }
  169. train_input_reader {
  170. label_map_path: "gs://tfstorage2/data/label_map.pbtxt"
  171. tf_record_input_reader {
  172. input_path: "gs://tfstorage2/data/train.record"
  173. }
  174. }
  175. eval_config {
  176. num_examples: 123
  177. metrics_set: "coco_detection_metrics"
  178. use_moving_averages: false
  179. }
  180. eval_input_reader {
  181. label_map_path: "gs://tfstorage2/data/label_map.pbtxt"
  182. shuffle: false
  183. num_epochs: 1
  184. num_readers: 1
  185. tf_record_input_reader {
  186. input_path: "gs://tfstorage2/data/val.record"
  187. }
  188. sample_1_of_n_examples: 123
  189. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement