Internal change

official/projects/detr/configs/detr.py

@@ -97,7 +97,7 @@ COCO_VAL_EXAMPLES = 5000
 @exp_factory.register_config_factory('detr_coco')
 def detr_coco() -> cfg.ExperimentConfig:
   """Config to get results that matches the paper."""
-  train_batch_size = 32
+  train_batch_size = 64
   eval_batch_size = 64
   steps_per_epoch = COCO_TRAIN_EXAMPLES // train_batch_size
   train_steps = 300 * steps_per_epoch  # 300 epochs

official/projects/detr/dataloaders/detr_input.py

@@ -46,10 +46,6 @@ class Parser(parser.Parser):
     #classes = data['groundtruth_classes'] + 1
     classes = data['groundtruth_classes']
     boxes = data['groundtruth_boxes']
-    # If not empty, `attributes` is a dict of (name, ground_truth) pairs.
-    # `ground_gruth` of attributes is assumed in shape [N, attribute_size].
-    # TODO(xianzhi): support parsing attributes weights.
-    attributes = data.get('groundtruth_attributes', {})
     is_crowd = data['groundtruth_is_crowd']
 
     # Gets original image.
@@ -139,10 +135,6 @@ class Parser(parser.Parser):
     groundtruths = {}
     classes = data['groundtruth_classes']
     boxes = data['groundtruth_boxes']
-    # If not empty, `attributes` is a dict of (name, ground_truth) pairs.
-    # `ground_gruth` of attributes is assumed in shape [N, attribute_size].
-    # TODO(xianzhi): support parsing attributes weights.
-    attributes = data.get('groundtruth_attributes', {})
     is_crowd = data['groundtruth_is_crowd']
 
     # Gets original image and its size.