diff --git a/demo/demo.ipynb b/demo/demo.ipynb index 1c4d74963..9c987497a 100644 --- a/demo/demo.ipynb +++ b/demo/demo.ipynb @@ -102,7 +102,7 @@ "from matplotlib import pyplot as plt\n", "from data import VOCDetection, VOCroot, AnnotationTransform\n", "# here we specify year (07 or 12) and dataset ('test', 'val', 'train') \n", - "testset = VOCDetection(VOCroot, [('2007', 'val')], None, AnnotationTransform)\n", + "testset = VOCDetection(VOCroot, [('2007', 'val')], None, AnnotationTransform())\n", "img_id = 60\n", "image = testset.pull_image(img_id)\n", "rgb_image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)\n",