Internal change

PiperOrigin-RevId: 486497693

official/vision/data/create_coco_tf_record.py

@@ -66,6 +66,9 @@ flags.DEFINE_boolean(
     'include_panoptic_masks', False, 'Whether to include category and '
     'instance masks in the result. These are required to run the PQ evaluator '
     'default: False.')
+flags.DEFINE_boolean(
+    'panoptic_skip_crowd', False, 'Whether to skip crowd or not for panoptic '
+    'annotations. default: False.')
 flags.DEFINE_string('output_file_prefix', '/tmp/train', 'Path to output file')
 flags.DEFINE_integer('num_shards', 32, 'Number of shards for output file.')
 _NUM_PROCESSES = flags.DEFINE_integer(
@@ -134,7 +137,9 @@ def generate_coco_panoptics_masks(segments_info, mask_path,
   for idx, segment in enumerate(segments_info):
     segment_id = segment['id']
     category_id = segment['category_id']
-
+    is_crowd = segment['iscrowd']
+    if FLAGS.panoptic_skip_crowd and is_crowd:
+      continue
     if is_category_thing[category_id]:
       encoded_category_id = _THING_CLASS_ID
       instance_id = idx + 1