Advertisement
Guest User

Untitled

a guest
Mar 21st, 2019
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.85 KB | None | 0 0
  1. model {
  2. ssd {
  3. num_classes: 1
  4. box_coder {
  5. faster_rcnn_box_coder {
  6. y_scale: 10.0
  7. x_scale: 10.0
  8. height_scale: 5.0
  9. width_scale: 5.0
  10. }
  11. }
  12. matcher {
  13. argmax_matcher {
  14. matched_threshold: 0.5
  15. unmatched_threshold: 0.5
  16. ignore_thresholds: false
  17. negatives_lower_than_unmatched: true
  18. force_match_for_each_row: true
  19. }
  20. }
  21. similarity_calculator {
  22. iou_similarity {
  23. }
  24. }
  25. anchor_generator {
  26. ssd_anchor_generator {
  27. num_layers: 6
  28. min_scale: 0.2
  29. max_scale: 0.95
  30. aspect_ratios: 1.0
  31. aspect_ratios: 2.0
  32. aspect_ratios: 0.5
  33. aspect_ratios: 3.0
  34. aspect_ratios: 0.3333
  35. }
  36. }
  37. image_resizer {
  38. fixed_shape_resizer {
  39. height: 300
  40. width: 300
  41. }
  42. }
  43. box_predictor {
  44. convolutional_box_predictor {
  45. min_depth: 0
  46. max_depth: 0
  47. num_layers_before_predictor: 0
  48. use_dropout: false
  49. dropout_keep_probability: 0.8
  50. kernel_size: 1
  51. box_code_size: 4
  52. apply_sigmoid_to_scores: false
  53. conv_hyperparams {
  54. activation: RELU_6,
  55. regularizer {
  56. l2_regularizer {
  57. weight: 0.00004
  58. }
  59. }
  60. initializer {
  61. truncated_normal_initializer {
  62. stddev: 0.03
  63. mean: 0.0
  64. }
  65. }
  66. batch_norm {
  67. train: true,
  68. scale: true,
  69. center: true,
  70. decay: 0.9997,
  71. epsilon: 0.001,
  72. }
  73. }
  74. }
  75. }
  76. feature_extractor {
  77. type: 'ssd_mobilenet_v1'
  78. min_depth: 16
  79. depth_multiplier: 1.0
  80. conv_hyperparams {
  81. activation: RELU_6,
  82. regularizer {
  83. l2_regularizer {
  84. weight: 0.00004
  85. }
  86. }
  87. initializer {
  88. truncated_normal_initializer {
  89. stddev: 0.03
  90. mean: 0.0
  91. }
  92. }
  93. batch_norm {
  94. train: true,
  95. scale: true,
  96. center: true,
  97. decay: 0.9997,
  98. epsilon: 0.001,
  99. }
  100. }
  101. }
  102. loss {
  103. classification_loss {
  104. weighted_sigmoid {
  105. anchorwise_output: true
  106. }
  107. }
  108. localization_loss {
  109. weighted_smooth_l1 {
  110. anchorwise_output: true
  111. }
  112. }
  113. hard_example_miner {
  114. num_hard_examples: 3000
  115. iou_threshold: 0.99
  116. loss_type: CLASSIFICATION
  117. max_negatives_per_positive: 3
  118. min_negatives_per_image: 0
  119. }
  120. classification_weight: 1.0
  121. localization_weight: 1.0
  122. }
  123. normalize_loss_by_num_matches: true
  124. post_processing {
  125. batch_non_max_suppression {
  126. score_threshold: 1e-8
  127. iou_threshold: 0.6
  128. max_detections_per_class: 100
  129. max_total_detections: 100
  130. }
  131. score_converter: SIGMOID
  132. }
  133. }
  134. }
  135. train_config: {
  136. batch_size: 2
  137. optimizer {
  138. rms_prop_optimizer: {
  139. learning_rate: {
  140. exponential_decay_learning_rate {
  141. initial_learning_rate: 0.004
  142. decay_steps: 800720
  143. decay_factor: 0.95
  144. }
  145. }
  146. momentum_optimizer_value: 0.9
  147. decay: 0.9
  148. epsilon: 1.0
  149. }
  150. }
  151. fine_tune_checkpoint: "ssd_mobilenet_v1_coco_2017_11_17/model.ckpt"
  152. from_detection_checkpoint: true
  153. num_steps: 100000
  154. data_augmentation_options {
  155. random_horizontal_flip {
  156. }
  157. }
  158. data_augmentation_options {
  159. ssd_random_crop {
  160. }
  161. }
  162. }
  163. train_input_reader: {
  164. tf_record_input_reader {
  165. input_path: "data/train.record"
  166. }
  167. label_map_path: "data/plate_number_map.pbtxt"
  168. }
  169. eval_config: {
  170. num_examples: 30
  171. max_evals: 10
  172. }
  173. eval_input_reader: {
  174. tf_record_input_reader {
  175. input_path: "data/test.record"
  176. }
  177. label_map_path: "data/plate_number_map.pbtxt"
  178. shuffle: false
  179. num_readers: 1
  180. num_epochs: 1
  181. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement