Commit cc93c45f authored by Zhichao Lu's avatar Zhichao Lu Committed by lzc5123016
Browse files

Added the necessary flags to the third party mask rcnn configs.

PiperOrigin-RevId: 185470191
parent 5e854f25
...@@ -129,7 +129,6 @@ train_input_reader: { ...@@ -129,7 +129,6 @@ train_input_reader: {
} }
eval_config: { eval_config: {
metrics_set: "coco_metrics"
use_moving_averages: false use_moving_averages: false
num_examples: 500 num_examples: 500
} }
......
...@@ -147,6 +147,8 @@ train_input_reader: { ...@@ -147,6 +147,8 @@ train_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
} }
eval_config: { eval_config: {
...@@ -161,6 +163,8 @@ eval_input_reader: { ...@@ -161,6 +163,8 @@ eval_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
shuffle: false shuffle: false
num_readers: 1 num_readers: 1
} }
...@@ -146,6 +146,8 @@ train_input_reader: { ...@@ -146,6 +146,8 @@ train_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
} }
eval_config: { eval_config: {
...@@ -160,6 +162,8 @@ eval_input_reader: { ...@@ -160,6 +162,8 @@ eval_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
shuffle: false shuffle: false
num_readers: 1 num_readers: 1
} }
...@@ -147,6 +147,8 @@ train_input_reader: { ...@@ -147,6 +147,8 @@ train_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
} }
eval_config: { eval_config: {
...@@ -161,6 +163,8 @@ eval_input_reader: { ...@@ -161,6 +163,8 @@ eval_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
shuffle: false shuffle: false
num_readers: 1 num_readers: 1
} }
...@@ -140,6 +140,7 @@ train_input_reader: { ...@@ -140,6 +140,7 @@ train_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/pet_train.record" input_path: "PATH_TO_BE_CONFIGURED/pet_train.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/pet_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/pet_label_map.pbtxt"
load_instance_masks: true
} }
eval_config: { eval_config: {
...@@ -154,6 +155,7 @@ eval_input_reader: { ...@@ -154,6 +155,7 @@ eval_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/pet_val.record" input_path: "PATH_TO_BE_CONFIGURED/pet_val.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/pet_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/pet_label_map.pbtxt"
load_instance_masks: true
shuffle: false shuffle: false
num_readers: 1 num_readers: 1
} }
...@@ -147,6 +147,8 @@ train_input_reader: { ...@@ -147,6 +147,8 @@ train_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_train.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
} }
eval_config: { eval_config: {
...@@ -161,6 +163,8 @@ eval_input_reader: { ...@@ -161,6 +163,8 @@ eval_input_reader: {
input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record" input_path: "PATH_TO_BE_CONFIGURED/mscoco_val.record"
} }
label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt" label_map_path: "PATH_TO_BE_CONFIGURED/mscoco_label_map.pbtxt"
load_instance_masks: true
mask_type: PNG_MASKS
shuffle: false shuffle: false
num_readers: 1 num_readers: 1
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment