Merge pull request #171 from Visual-Behavior/Daria-minor-corrections

Fixed some issues
Visual-Behavior · Apr 21, 2022 · 7b73560 · 7b73560
2 parents 22c8be5 + 063ac34
commit 7b73560
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/docsource/source/getting_started/alonet.ipynb b/docsource/source/getting_started/alonet.ipynb
@@ -105,7 +105,7 @@
     "        )(frame)\n",
     "        return frame.norm_resnet()\n",
     "\n",
-    "    def val_transform(self, frame):\n",
+    "    def val_transform(self, frame, same_on_sequence: bool = True, same_on_frames: bool = False):\n",
     "        frame = T.RandomResizeWithAspectRatio(\n",
     "            [800], max_size=1333, same_on_sequence=same_on_sequence, same_on_frames=same_on_frames\n",
     "        )(frame)\n",
@@ -116,11 +116,13 @@
     "            # Setup train/val loaders\n",
     "            self.coco_train = alodataset.CocoBaseDataset(\n",
     "                img_folder = \"train2017\",\n",
-    "                ann_file = \"annotations/instances_val2017.json\"\n",
+    "                ann_file = \"annotations/instances_train2017.json\",\n"
+    "                transform_fn=self.train_transform \n"
     "            )\n",
     "            self.coco_val = alodataset.CocoBaseDataset(\n",
     "                img_folder = \"val2017\",\n",
     "                ann_file = \"annotations/instances_val2017.json\"\n",
+    "                transform_fn=self.val_transform \n"
     "            )\n",
     "\n",
     "    def train_dataloader(self):\n",