model { faster_rcnn { num_classes: 90 image_resizer { keep_aspect_ratio_resizer { min_dimension: 600 max_dimension: 1024 } } feature_extractor { type: "faster_rcnn_inception_resnet_v2" first_stage_features_stride: 8 } first_stage_anchor_generator { grid_anchor_generator { height_stride: 8 width_stride: 8 scales: 0.25 scales: 0.5 scales: 1.0 scales: 2.0 aspect_ratios: 0.5 aspect_ratios: 1.0 aspect_ratios: 2.0 } } first_stage_atrous_rate: 2 first_stage_box_predictor_conv_hyperparams { op: CONV regularizer { l2_regularizer { weight: 0.0 } } initializer { truncated_normal_initializer { stddev: 0.00999999977648 } } } first_stage_nms_score_threshold: 0.0 first_stage_nms_iou_threshold: 0.699999988079 first_stage_max_proposals: 300 first_stage_localization_loss_weight: 2.0 first_stage_objectness_loss_weight: 1.0 initial_crop_size: 17 maxpool_kernel_size: 1 maxpool_stride: 1 second_stage_box_predictor { mask_rcnn_box_predictor { fc_hyperparams { op: FC regularizer { l2_regularizer { weight: 0.0 } } initializer { variance_scaling_initializer { factor: 1.0 uniform: true mode: FAN_AVG } } } use_dropout: false dropout_keep_probability: 1.0 } } second_stage_post_processing { batch_non_max_suppression { score_threshold: 0.0 iou_threshold: 0.600000023842 max_detections_per_class: 100 max_total_detections: 100 } score_converter: SOFTMAX } second_stage_localization_loss_weight: 2.0 second_stage_classification_loss_weight: 1.0 } } train_config { batch_size: 1 data_augmentation_options { random_adjust_contrast { min_delta: 0.899999976158 max_delta: 1.10000002384 } } keep_checkpoint_every_n_hours: 1 optimizer { momentum_optimizer { learning_rate { manual_step_learning_rate { initial_learning_rate: 0.000300000014249 schedule { step: 300000 learning_rate: 2.99999992421e-05 } schedule { step: 400000 learning_rate: 3.00000010611e-06 } } } momentum_optimizer_value: 0.899999976158 } use_moving_average: false } gradient_clipping_by_norm: 10.0 fine_tune_checkpoint: "/ai4efs/models/object_detection/faster_rcnn_inception_resnet_v2_atrous/faster_rcnn_inception_resnet_v2_atrous_coco_2018_01_28/model.ckpt" from_detection_checkpoint: true num_steps: 200000 } train_input_reader { label_map_path: "/data/tfrecords/caltechcameratraps/oneclass/oneclass_label_map.pbtxt" tf_record_input_reader { input_path: "/data/tfrecords/megadetector_2/train-?????-of-?????" } max_number_of_boxes: 200 } eval_config { num_examples: 1725 eval_interval_secs: 3600 metrics_set: "coco_detection_metrics" use_moving_averages: false } eval_input_reader { label_map_path: "/data/tfrecords/caltechcameratraps/oneclass/oneclass_label_map.pbtxt" shuffle: false num_readers: 1 tf_record_input_reader { input_path: "/data/tfrecords/megadetector_2/val-?????-of-?????" } }