Advertisement
Bart91

Config file

Dec 2nd, 2018
65
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
XML 4.60 KB | None | 0 0
  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_v2"
  12.           depth_multiplier: 1.0
  13.           min_depth: 16
  14.           conv_hyperparams {
  15.             regularizer {
  16.               l2_regularizer {
  17.                 weight: 3.99999989895e-05
  18.               }
  19.             }
  20.             initializer {
  21.               truncated_normal_initializer {
  22.                 mean: 0.0
  23.                 stddev: 0.0299999993294
  24.               }
  25.             }
  26.             activation: RELU_6
  27.             batch_norm {
  28.                 decay: 0.999700009823
  29.                 center: true
  30.                 scale: true
  31.                 epsilon: 0.0010000000475
  32.                 train: true
  33.               }
  34.           }
  35.           use_depthwise: 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.           }
  53.         }
  54.         box_predictor {
  55.           convolutional_box_predictor {
  56.             conv_hyperparams {
  57.               regularizer {
  58.                 l2_regularizer {
  59.                   weight: 3.99999989895e-05
  60.                 }
  61.               }
  62.               initializer {
  63.                 truncated_normal_initializer {
  64.                   mean: 0.0
  65.                   stddev: 0.0299999993294
  66.                 }
  67.               }
  68.               activation: RELU_6
  69.               batch_norm {
  70.                 decay: 0.999700009823
  71.                 center: true
  72.                 scale: true
  73.                 epsilon: 0.0010000000475
  74.                 train: true
  75.               }
  76.             }
  77.             min_depth: 0
  78.             max_depth: 0
  79.             num_layers_before_predictor: 0
  80.             use_dropout: false
  81.             dropout_keep_probability: 0.800000011921
  82.             kernel_size: 3
  83.             box_code_size: 4
  84.             apply_sigmoid_to_scores: false
  85.           }
  86.         }
  87.         anchor_generator {
  88.           ssd_anchor_generator {
  89.             num_layers: 6
  90.             min_scale: 0.20000000298
  91.             max_scale: 0.949999988079
  92.             aspect_ratios: 1.0
  93.             aspect_ratios: 2.0
  94.             aspect_ratios: 0.5
  95.             aspect_ratios: 3.0
  96.             aspect_ratios: 0.333299994469
  97.           }
  98.         }
  99.         post_processing {
  100.           batch_non_max_suppression {
  101.             score_threshold: 0.300000011921
  102.             iou_threshold: 0.600000023842
  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.           hard_example_miner {
  110.             num_hard_examples: 3000
  111.             iou_threshold: 0.990000009537
  112.             loss_type: CLASSIFICATION
  113.             max_negatives_per_positive: 3
  114.             min_negatives_per_image: 3
  115.           }
  116.           classification_weight: 1.0
  117.           localization_weight: 1.0
  118.         }
  119.       }
  120.     }
  121.     train_config {
  122.       batch_size: 12
  123.       optimizer {
  124.         rms_prop_optimizer {
  125.           learning_rate {
  126.             exponential_decay_learning_rate {
  127.               initial_learning_rate: 0.00400000018999
  128.               decay_steps: 800720
  129.               decay_factor: 0.949999988079
  130.             }
  131.           }
  132.           momentum_optimizer_value: 0.899999976158
  133.           decay: 0.899999976158
  134.           epsilon: 1.0
  135.         }
  136.       }
  137.       fine_tune_checkpoint: "/home/bartek/tensorflow/models/research/object_detection/ssd_mobilenet_v2_coco_2018_03_29/model.ckpt"
  138.       num_steps: 200000
  139.       fine_tune_checkpoint_type: "detection"
  140.     }
  141.     train_input_reader {
  142.       label_map_path: "/home/bartek/Downloads/person/data/label_map.pbtxt"
  143.       tf_record_input_reader {
  144.         input_path: "/home/bartek/Downloads/person/data/train.record"
  145.       }
  146.     }
  147.     eval_config {
  148.       num_examples: 8000
  149.       max_evals: 10
  150.       use_moving_averages: false
  151.     }
  152.     eval_input_reader {
  153.       label_map_path: "/home/bartek/Downloads/person/data/label_map.pbtxt"
  154.       shuffle: false
  155.       num_readers: 1
  156.       tf_record_input_reader {
  157.         input_path: "/home/bartek/Downloads/person/data/test.record"
  158.       }
  159.     }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement