Myosotis-Researches 0.1.24__py3-none-any.whl → 0.1.26__py3-none-any.whl

myosotis_researches/CcGAN/utils/__init__.py

@@ -1,2 +1,2 @@
 from .concat_image import concat_image
-from .view_dataset import view_dataset
+from .dataset import view_dataset, download_datasets

myosotis_researches/CcGAN/utils/dataset.py

@@ -0,0 +1,45 @@
+import gdown
+import getpass
+import h5py
+import os
+import subprocess
+
+
+# view_dataset
+def _print_hdf5(name, obj):
+    indent = "  " * name.count("/")
+    if isinstance(obj, h5py.Dataset):
+        print(f"{indent}[Dataset] {name} shape={obj.shape} dtype={obj.dtype}")
+    elif isinstance(obj, h5py.Group):
+        print(f"{indent}[Group]   {name}")
+
+
+def view_dataset(dataset_path):
+    with h5py.File(dataset_path, "r") as f:
+        f.visititems(_print_hdf5)
+
+
+# download_datasets
+_file_id_dict = {
+    "Ra": "1YVLFAaQ1ldre0ASLSRdV_kPIcAMkwGqO",
+    "MNIST": "18249ryrDPsznWWtOkJuALZWCVoZqfibC",
+}
+
+
+def download_datasets(datasets_name, datasets_dir):
+    if not os.path.exists(datasets_dir):
+        os.makedirs(datasets_dir, exist_ok=True)
+    zipped_datasets_path = os.path.join(datasets_dir, f"{datasets_name}.rar")
+    file_id = _file_id_dict[datasets_name]
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, zipped_datasets_path, quiet=False, use_cookies=False)
+    unzip_password = getpass.getpass("Please input the zipped file's password:")
+    try:
+        cmd = ["unrar", "x", f"-p{unzip_password}", zipped_datasets_path, datasets_dir]
+        subprocess.run(cmd, check=True)
+    except subprocess.CalledProcessError as e:
+        print(e)
+    os.remove(zipped_datasets_path)
+
+
+__all__ = ["view_dataset", "download_datasets"]

{myosotis_researches-0.1.24.dist-info → myosotis_researches-0.1.26.dist-info}/METADATA

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

{myosotis_researches-0.1.24.dist-info → myosotis_researches-0.1.26.dist-info}/RECORD

@@ -26,11 +26,11 @@ myosotis_researches/CcGAN/train/train_ccgan.py,sha256=0Qxibgd2-WaYgbyYeeOyiMkdcw
 myosotis_researches/CcGAN/train/train_cgan.py,sha256=sxMzvlmdjmqufwJFxBwatcoJecYqn2Uidedu15CL9ws,9619
 myosotis_researches/CcGAN/train/train_cgan_concat.py,sha256=OrQbwdU_ujUeKFGixUUpnini6rURtbuHv9NDrP6g0X0,8861
 myosotis_researches/CcGAN/train/train_net_for_label_embed.py,sha256=4j6r4_o4rXgAN4MdUQL-TXqZJpbhH7d9gWQR8YzBlXw,6976
-myosotis_researches/CcGAN/utils/__init__.py,sha256=q13-OwYs-mkVz3qdJ2tNFktlulHNXx0BR13bY0rN1V0,78
+myosotis_researches/CcGAN/utils/__init__.py,sha256=Y6HFAJRAOa4rMXhd1KIyr3NL_GzO0vjbp-dE4NT7QW8,92
 myosotis_researches/CcGAN/utils/concat_image.py,sha256=BIGKz52Inn9S7M5fBFKye2V9bLJ0DqEQILoOVWAXUiE,2165
-myosotis_researches/CcGAN/utils/view_dataset.py,sha256=BaAKII7BAItVhL--H5q88H9cUR7nSDTtAvV_uz6b6hs,415
-myosotis_researches-0.1.24.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.1.24.dist-info/METADATA,sha256=s83HKvH4e0B4DbBXF1sUe6wHEg8N3PZbFmf5h4xaP6o,3484
-myosotis_researches-0.1.24.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-myosotis_researches-0.1.24.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.1.24.dist-info/RECORD,,
+myosotis_researches/CcGAN/utils/dataset.py,sha256=wDe-fABslaLXOY5Kbmlm0mQ_8CPEK7Y2YkEy-XAiQHI,1363
+myosotis_researches-0.1.26.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+myosotis_researches-0.1.26.dist-info/METADATA,sha256=yN5d9W51yJkKBOGW2VrFdBVQx5jSFMrkmai57ZuoqGM,3484
+myosotis_researches-0.1.26.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
+myosotis_researches-0.1.26.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
+myosotis_researches-0.1.26.dist-info/RECORD,,

myosotis_researches/CcGAN/utils/view_dataset.py

@@ -1,16 +0,0 @@
-import h5py
-
-
-def _print_hdf5(name, obj):
-    indent = "  " * name.count("/")
-    if isinstance(obj, h5py.Dataset):
-        print(f"{indent}[Dataset] {name} shape={obj.shape} dtype={obj.dtype}")
-    elif isinstance(obj, h5py.Group):
-        print(f"{indent}[Group]   {name}")
-
-
-def view_dataset(dataset_path):
-    with h5py.File(dataset_path, "r") as f:
-        f.visititems(_print_hdf5)
-
-__all__ = ["view_dataset"]