Merge pull request #64 from tilt/master

Minor fix in demo notebook
amdegroot · Oct 25, 2017 · 197f922 · 197f922
2 parents bc3bc6c + a813db9
commit 197f922
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",