diff --git a/data/coco.py b/data/coco.py index 765531761..4f3ac26e9 100644 --- a/data/coco.py +++ b/data/coco.py @@ -44,7 +44,7 @@ class COCOAnnotationTransform(object): Initilized with a dictionary lookup of classnames to indexes """ def __init__(self): - self.label_map = get_label_map(osp.join(COCO_ROOT, 'coco_labels.txt')) + self.label_map = get_label_map('data/coco_labels.txt') def __call__(self, target, width, height): """