MobileViT-XS tutorial and MCT-model garden initial commit

sony · Apr 24, 2024 · 4d0d734 · 4d0d734
1 parent 9ae001e
commit 4d0d734
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tutorials/notebooks/pytorch/ptq/pytorch_mobilevit_xs_for_imx500.ipynb b/tutorials/notebooks/pytorch/ptq/pytorch_mobilevit_xs_for_imx500.ipynb
@@ -101,8 +101,12 @@
     "!mkdir imagenet\n",
     "!mkdir imagenet/val\n",
     "!wget https://image-net.org/data/ILSVRC/2012/ILSVRC2012_img_val.tar\n",
-    "!mv ILSVRC2012_img_val.tar imagenet/\n",
-    "!tar -xvf imagenet/ILSVRC2012_img_val.tar -C imagenet/val\n",
+    "!mv ILSVRC2012_img_val.tar imagenet/val\n",
+    "!tar -xvf imagenet/val/ILSVRC2012_img_val.tar -C imagenet/val\n",
+    "!rm -f imagenet/val/ILSVRC2012_img_val.tar\n",
+    "# We follow https://github.com/pytorch/examples/blob/main/imagenet/extract_ILSVRC.sh here\n",
+    "# get script from soumith and run; this script creates all class directories and moves images into corresponding directories\n",
+    "!cd imagenet/val/ && wget -qO- https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/valprep.sh | bash\n",
     "!echo Done loading ILSVRC2012_img_val images"
    ]
   },
@@ -167,8 +171,9 @@
     "data_config = resolve_data_config(args=DATA_ARGS,\n",
     "                                  model=model)\n",
     "\n",
-    "val_dataset = create_dataset(name=DATASET, root=REPRESENTATIVE_DATASET_FOLDER, is_training=False,\n",
-    "                             batch_size=BATCH_SIZE)\n",
+    "val_dataset = create_dataset(name=DATASET, root=REPRESENTATIVE_DATASET_FOLDER,\n",
+    "                             is_training=False, batch_size=BATCH_SIZE)\n",
+    "\n",
     "representative_dataset = create_loader(\n",
     "    val_dataset,\n",
     "    input_size=data_config['input_size'],\n",