"git@developer.sourcefind.cn:OpenDAS/torch-scatter.git" did not exist on "f05c8c540cf36a00f1c316a7fc771a37ab0144b8"
Commit 2624a23e authored by Zhichao Lu's avatar Zhichao Lu Committed by pkulzc
Browse files

Minor fixes to match git repo.

PiperOrigin-RevId: 192166224
parent e25c014e
...@@ -104,21 +104,22 @@ def dict_to_tf_example(data, ...@@ -104,21 +104,22 @@ def dict_to_tf_example(data,
truncated = [] truncated = []
poses = [] poses = []
difficult_obj = [] difficult_obj = []
for obj in data['object']: if data.has_key('object'):
difficult = bool(int(obj['difficult'])) for obj in data['object']:
if ignore_difficult_instances and difficult: difficult = bool(int(obj['difficult']))
continue if ignore_difficult_instances and difficult:
continue
difficult_obj.append(int(difficult))
difficult_obj.append(int(difficult))
xmin.append(float(obj['bndbox']['xmin']) / width)
ymin.append(float(obj['bndbox']['ymin']) / height) xmin.append(float(obj['bndbox']['xmin']) / width)
xmax.append(float(obj['bndbox']['xmax']) / width) ymin.append(float(obj['bndbox']['ymin']) / height)
ymax.append(float(obj['bndbox']['ymax']) / height) xmax.append(float(obj['bndbox']['xmax']) / width)
classes_text.append(obj['name'].encode('utf8')) ymax.append(float(obj['bndbox']['ymax']) / height)
classes.append(label_map_dict[obj['name']]) classes_text.append(obj['name'].encode('utf8'))
truncated.append(int(obj['truncated'])) classes.append(label_map_dict[obj['name']])
poses.append(obj['pose'].encode('utf8')) truncated.append(int(obj['truncated']))
poses.append(obj['pose'].encode('utf8'))
example = tf.train.Example(features=tf.train.Features(feature={ example = tf.train.Example(features=tf.train.Features(feature={
'image/height': dataset_util.int64_feature(height), 'image/height': dataset_util.int64_feature(height),
......
...@@ -31,7 +31,7 @@ while for the Weighted PASCAL VOC metric the final mAP value will be influenced ...@@ -31,7 +31,7 @@ while for the Weighted PASCAL VOC metric the final mAP value will be influenced
Similar to pascal voc 2007 detection metric, but computes the intersection over Similar to pascal voc 2007 detection metric, but computes the intersection over
union based on the object masks instead of object boxes. union based on the object masks instead of object boxes.
## Weighted PASCAL VOC detection metric ## Weighted PASCAL VOC instance segmentation metric
`EvalConfig.metrics_set='weighted_pascal_voc_instance_segmentation_metrics'` `EvalConfig.metrics_set='weighted_pascal_voc_instance_segmentation_metrics'`
......
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