172 lines
3.9 KiB
Plaintext
172 lines
3.9 KiB
Plaintext
model {
|
|
ssd {
|
|
num_classes: 90
|
|
image_resizer {
|
|
fixed_shape_resizer {
|
|
height: 320
|
|
width: 320
|
|
}
|
|
}
|
|
feature_extractor {
|
|
type: "ssd_mobilenet_v2"
|
|
depth_multiplier: 0.5
|
|
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.97000002861
|
|
center: true
|
|
scale: true
|
|
epsilon: 0.0010000000475
|
|
train: true
|
|
}
|
|
}
|
|
use_depthwise: true
|
|
override_base_feature_extractor_hyperparams: 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
|
|
use_matmul_gather: true
|
|
}
|
|
}
|
|
similarity_calculator {
|
|
iou_similarity {
|
|
}
|
|
}
|
|
box_predictor {
|
|
convolutional_box_predictor {
|
|
conv_hyperparams {
|
|
regularizer {
|
|
l2_regularizer {
|
|
weight: 3.99999989895e-05
|
|
}
|
|
}
|
|
initializer {
|
|
random_normal_initializer {
|
|
mean: 0.0
|
|
stddev: 0.0299999993294
|
|
}
|
|
}
|
|
activation: RELU_6
|
|
batch_norm {
|
|
decay: 0.97000002861
|
|
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
|
|
class_prediction_bias_init: -4.59999990463
|
|
use_depthwise: true
|
|
}
|
|
}
|
|
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: 9.99999993923e-09
|
|
iou_threshold: 0.600000023842
|
|
max_detections_per_class: 100
|
|
max_total_detections: 100
|
|
use_static_shapes: true
|
|
}
|
|
score_converter: SIGMOID
|
|
}
|
|
normalize_loss_by_num_matches: true
|
|
loss {
|
|
localization_loss {
|
|
weighted_smooth_l1 {
|
|
delta: 1.0
|
|
}
|
|
}
|
|
classification_loss {
|
|
weighted_sigmoid_focal {
|
|
gamma: 2.0
|
|
alpha: 0.75
|
|
}
|
|
}
|
|
classification_weight: 1.0
|
|
localization_weight: 1.0
|
|
}
|
|
encode_background_as_zeros: true
|
|
normalize_loc_loss_by_codesize: true
|
|
inplace_batchnorm_update: true
|
|
freeze_batchnorm: false
|
|
}
|
|
}
|
|
train_config {
|
|
batch_size: 512
|
|
data_augmentation_options {
|
|
random_horizontal_flip {
|
|
}
|
|
}
|
|
data_augmentation_options {
|
|
ssd_random_crop {
|
|
}
|
|
}
|
|
sync_replicas: true
|
|
optimizer {
|
|
momentum_optimizer {
|
|
learning_rate {
|
|
cosine_decay_learning_rate {
|
|
learning_rate_base: 0.40000000596
|
|
total_steps: 400000
|
|
warmup_learning_rate: 0.133330002427
|
|
warmup_steps: 2000
|
|
}
|
|
}
|
|
momentum_optimizer_value: 0.899999976158
|
|
}
|
|
use_moving_average: false
|
|
}
|
|
fine_tune_checkpoint: "checkpoint/to/fine/tune/from/if/needed"
|
|
num_steps: 400000
|
|
startup_delay_steps: 0.0
|
|
replicas_to_aggregate: 8
|
|
max_number_of_boxes: 100
|
|
unpad_groundtruth_tensors: false
|
|
use_bfloat16: false
|
|
}
|