Myosotis-Researches 0.0.20__tar.gz → 0.0.22__tar.gz

{myosotis_researches-0.0.20 → myosotis_researches-0.0.22/Myosotis_Researches.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.20
+Version: 0.0.22
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie

{myosotis_researches-0.0.20 → myosotis_researches-0.0.22}/Myosotis_Researches.egg-info/SOURCES.txt

@@ -51,5 +51,6 @@ myosotis_researches/CcGAN/train_128_output_10/utils.py
 myosotis_researches/CcGAN/utils/__init__.py
 myosotis_researches/CcGAN/utils/concat_image_horizontal.py
 myosotis_researches/CcGAN/utils/concat_image_vertical.py
+myosotis_researches/CcGAN/utils/install_datasets.py
 myosotis_researches/CcGAN/utils/make_h5.py
 myosotis_researches/CcGAN/utils/print_hdf5_structure.py

{myosotis_researches-0.0.20/Myosotis_Researches.egg-info → myosotis_researches-0.0.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.20
+Version: 0.0.22
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie

{myosotis_researches-0.0.20 → myosotis_researches-0.0.22}/myosotis_researches/CcGAN/utils/__init__.py

@@ -2,5 +2,6 @@ from .print_hdf5_structure import print_hdf5_structure
 from .concat_image_horizontal import concat_image_horizontal
 from .concat_image_vertical import concat_image_vertical
 from .make_h5 import make_h5
+from .install_datasets import install_datasets
 
-__all__ = ["print_hdf5_structure", "concat_image_horizontal", "concat_image_vertical", "make_h5"]
+__all__ = ["print_hdf5_structure", "concat_image_horizontal", "concat_image_vertical", "make_h5", "install_datasets"]

myosotis_researches-0.0.22/myosotis_researches/CcGAN/utils/install_datasets.py

@@ -0,0 +1,43 @@
+import gdown
+from importlib import resources
+import os
+import rarfile
+
+# Paths
+datasets_dir = resources.files("myosotis_researches").joinpath("CcGAN", "datasets")
+
+# File ID dictionary
+file_id_dict = {
+    "Ra": "1DJfpvXTX1x6Ldf4nZzqzlVsQSx9iXo9I",
+    "MNIST": "1ZmTSX_wl68nsQI0ImfhpIsPClW_QrK1T",
+}
+
+
+# Function
+def install_datasets(datasets_name):
+
+    # Path
+    zipped_dataset_path = os.path.join(datasets_dir, f"{datasets_name}_datasets.rar")
+
+    # File ID
+    file_id = file_id_dict[datasets_name]
+
+    # URL
+    url = f"https://drive.google.com/uc?id={file_id}"
+
+    # Download
+    gdown.download(url, zipped_dataset_path, quiet=False, use_cookies=False)
+
+    # Unzip
+    with rarfile.RarFile(zipped_dataset_path) as rf:
+        if rf.needs_password():
+            unzip_password = input("Password: ")
+            rf.extractall(path=datasets_dir, pwd=unzip_password)
+        else:
+            rf.extractall(path=datasets_dir)
+
+    # Delete zipped datasets
+    os.remove(zipped_dataset_path)
+
+
+__all__ = ["install_datasets"]

{myosotis_researches-0.0.20 → myosotis_researches-0.0.22}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="Myosotis-Researches",
-    version="0.0.20",
+    version="0.0.22",
     description="A repository for storing my progress of researches.",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",