fetch hymenoptera_data

Author: brownj85

demonstrations_v2/tutorial_quantum_transfer_learning/demo.py

@@ -123,6 +123,8 @@
 import time
 import os
 import copy
+import urllib.request
+import shutil
 
 # PyTorch
 import torch
@@ -211,7 +213,13 @@
     ),
 }
 
-data_dir = "../_data/hymenoptera_data"
+data_dir = "hymenoptera_data"
+if not os.path.exists(data_dir):
+    urllib.request.urlretrieve(
+        "https://download.pytorch.org/tutorial/hymenoptera_data.zip", f"{data_dir}.zip"
+    )
+    shutil.unpack_archive(f"{data_dir}.zip")
+
 image_datasets = {
     x if x == "train" else "validation": datasets.ImageFolder(
         os.path.join(data_dir, x), data_transforms[x]
@@ -609,4 +617,4 @@ def visualize_model(model, num_images=6, fig_name="Predictions"):
 #
 # About the author
 # ----------------
-#
+#