Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- model {
- ssd {
- num_classes: 1
- image_resizer {
- fixed_shape_resizer {
- height: 300
- width: 300
- }
- }
- feature_extractor {
- type: "ssd_mobilenet_v2"
- depth_multiplier: 1.0
- min_depth: 16
- conv_hyperparams {
- regularizer {
- l2_regularizer {
- weight: 3.99999989895e-05
- }
- }
- initializer {
- truncated_normal_initializer {
- mean: 0.0
- stddev: 0.0299999993294
- }
- }
- activation: RELU_6
- batch_norm {
- decay: 0.999700009823
- center: true
- scale: true
- epsilon: 0.0010000000475
- train: true
- }
- }
- use_depthwise: true
- }
- box_coder {
- faster_rcnn_box_coder {
- y_scale: 10.0
- x_scale: 10.0
- height_scale: 5.0
- width_scale: 5.0
- }
- }
- matcher {
- argmax_matcher {
- matched_threshold: 0.5
- unmatched_threshold: 0.5
- ignore_thresholds: false
- negatives_lower_than_unmatched: true
- force_match_for_each_row: true
- }
- }
- box_predictor {
- convolutional_box_predictor {
- conv_hyperparams {
- regularizer {
- l2_regularizer {
- weight: 3.99999989895e-05
- }
- }
- initializer {
- truncated_normal_initializer {
- mean: 0.0
- stddev: 0.0299999993294
- }
- }
- activation: RELU_6
- batch_norm {
- decay: 0.999700009823
- center: true
- scale: true
- epsilon: 0.0010000000475
- train: true
- }
- }
- min_depth: 0
- max_depth: 0
- num_layers_before_predictor: 0
- use_dropout: false
- dropout_keep_probability: 0.800000011921
- kernel_size: 3
- box_code_size: 4
- apply_sigmoid_to_scores: false
- }
- }
- anchor_generator {
- ssd_anchor_generator {
- num_layers: 6
- min_scale: 0.20000000298
- max_scale: 0.949999988079
- aspect_ratios: 1.0
- aspect_ratios: 2.0
- aspect_ratios: 0.5
- aspect_ratios: 3.0
- aspect_ratios: 0.333299994469
- }
- }
- post_processing {
- batch_non_max_suppression {
- score_threshold: 0.300000011921
- iou_threshold: 0.600000023842
- max_detections_per_class: 100
- max_total_detections: 100
- }
- score_converter: SIGMOID
- }
- normalize_loss_by_num_matches: true
- hard_example_miner {
- num_hard_examples: 3000
- iou_threshold: 0.990000009537
- loss_type: CLASSIFICATION
- max_negatives_per_positive: 3
- min_negatives_per_image: 3
- }
- classification_weight: 1.0
- localization_weight: 1.0
- }
- }
- }
- train_config {
- batch_size: 12
- optimizer {
- rms_prop_optimizer {
- learning_rate {
- exponential_decay_learning_rate {
- initial_learning_rate: 0.00400000018999
- decay_steps: 800720
- decay_factor: 0.949999988079
- }
- }
- momentum_optimizer_value: 0.899999976158
- decay: 0.899999976158
- epsilon: 1.0
- }
- }
- fine_tune_checkpoint: "/home/bartek/tensorflow/models/research/object_detection/ssd_mobilenet_v2_coco_2018_03_29/model.ckpt"
- num_steps: 200000
- fine_tune_checkpoint_type: "detection"
- }
- train_input_reader {
- label_map_path: "/home/bartek/Downloads/person/data/label_map.pbtxt"
- tf_record_input_reader {
- input_path: "/home/bartek/Downloads/person/data/train.record"
- }
- }
- eval_config {
- num_examples: 8000
- max_evals: 10
- use_moving_averages: false
- }
- eval_input_reader {
- label_map_path: "/home/bartek/Downloads/person/data/label_map.pbtxt"
- shuffle: false
- num_readers: 1
- tf_record_input_reader {
- input_path: "/home/bartek/Downloads/person/data/test.record"
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement