Skip to content

S
seminar-breakout
Commit 22f410e2 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

[MaskRCNN] fix errors in logging

parent b419cc81
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
examples/FasterRCNN/dataset/coco.py
View file @ 22f410e2
...@@ -110,7 +110,8 @@ class COCODetection(DatasetSplit): ...@@ -110,7 +110,8 @@ class COCODetection(DatasetSplit):
""" """
if add_mask: if add_mask:
assert add_gt 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 = self.coco.getImgIds()
img_ids.sort() img_ids.sort()
# list of dict, each has keys: height,width,id,file_name # 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