Commit 22f410e2 authored by Yuxin Wu's avatar Yuxin Wu

[MaskRCNN] fix errors in logging

parent b419cc81
......@@ -110,7 +110,8 @@ class COCODetection(DatasetSplit):
"""
if add_mask:
assert add_gt
with timed_operation('Load Groundtruth Boxes for {}'.format(self.name)):
with timed_operation('Load Groundtruth Boxes for {}'.format(
os.path.basename(self.annotation_file))):
img_ids = self.coco.getImgIds()
img_ids.sort()
# list of dict, each has keys: height,width,id,file_name
......
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