forked from yekeren/Cap2Det
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcoco17_word_vector_match.pbtxt
137 lines (136 loc) · 3.06 KB
/
coco17_word_vector_match.pbtxt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
train_reader {
cap2det_reader {
input_pattern: "raw-data-coco/coco17_train.record*"
interleave_cycle_length: 1
is_training: true
shuffle_buffer_size: 2000
map_num_parallel_calls: 10
prefetch_buffer_size: 500
batch_size: 2
image_resizer {
keep_aspect_ratio_resizer {
min_dimension: 1000
}
}
preprocess_options {
random_flip_left_right_prob: 0.5
}
max_num_proposals: 500
batch_resize_scale_value: 1.2
batch_resize_scale_value: 0.8
batch_resize_scale_value: 0.6
batch_resize_scale_value: 0.4
}
}
eval_reader {
cap2det_reader {
input_pattern: "raw-data-coco/coco17_val.record*"
interleave_cycle_length: 1
is_training: false
shuffle_buffer_size: 500
map_num_parallel_calls: 5
prefetch_buffer_size: 500
batch_size: 1
image_resizer {
keep_aspect_ratio_resizer {
min_dimension: 1000
}
}
max_num_proposals: 500
}
}
model {
[Cap2DetModel.ext] {
midn_loss_weight: 1.0
oicr_loss_weight: 0.5
frcnn_options {
feature_extractor {
type: 'faster_rcnn_inception_v2'
first_stage_features_stride: 16
}
initial_crop_size: 14
maxpool_kernel_size: 2
maxpool_stride: 2
dropout_keep_prob: 0.5
dropout_on_feature_map: false
checkpoint_path: 'zoo/inception_v2_2016_08_28/inception_v2.ckpt'
}
fc_hyperparams {
op: FC
activation: RELU_6
regularizer {
l2_regularizer {
weight: 0.000001
}
}
initializer {
truncated_normal_initializer {
mean: 0.0
stddev: 0.01
}
}
}
oicr_iterations: 3
oicr_iou_threshold: 0.6
midn_post_processor {
score_thresh: 0.00001
iou_thresh: 0.4
max_size_per_class: 100
max_total_size: 300
}
oicr_post_processor {
score_thresh: 0.00001
iou_thresh: 0.3
max_size_per_class: 100
max_total_size: 300
}
eval_min_dimension: 1200
eval_min_dimension: 800
eval_min_dimension: 600
eval_min_dimension: 400
oicr_use_proba_r_given_c: true
label_extractor {
word_vector_match_extractor {
label_file: 'data/coco_label.txt'
open_vocabulary_file: 'data/coco_open_vocab.txt'
open_vocabulary_word_embedding_file: 'data/coco_open_vocab_300d.npy'
}
}
}
}
train_config {
max_steps: 500000
learning_rate: 0.01
learning_rate_decay {
decay_steps: 200000
decay_rate: 1.0
staircase: true
}
moving_average_decay: 0.0
optimizer {
adagrad {
}
}
gradient_multiplier {
scope: 'first_stage_feature_extraction'
multiplier: 0.0
}
gradient_multiplier {
scope: 'second_stage_feature_extraction'
multiplier: 1.0
}
gradient_multiplier {
scope: 'first_stage_feature_extraction/InceptionV2/Mixed_4e'
multiplier: 1.0
}
sync_replicas: false
save_summary_steps: 2000
save_checkpoints_steps: 2000
keep_checkpoint_max: 10
log_step_count_steps: 10
}
eval_config {
steps: 100
start_delay_secs: 0
throttle_secs: 60
}