Myosotis-Researches 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

myosotis_researches/CcGAN/utils/concat_image.py

@@ -62,9 +62,9 @@ def _concat_image_vertical(img_list: list[Image], gap=2) -> Image:
 def concat_image(img_list: list[Image], gap=2, direction="vertical"):
 
     if direction == "vertical":
-        _concat_image_vertical(img_list, gap=gap)
+        return _concat_image_vertical(img_list, gap=gap)
     elif direction == "horizontal":
-        _concat_image_horizontal(img_list, gap=gap)
+        return _concat_image_horizontal(img_list, gap=gap)
     else:
         raise ValueError("Direction should be 'vertical' or 'horizontal'.")
 

myosotis_researches/CcGAN/utils/make_h5.py

@@ -1,43 +1,34 @@
 import h5py
-import numpy as np
+from importlib import resources
 import os
-from PIL import Image
-from .print_hdf5 import print_hdf5
-
-# Make all images to a HDF5 file
-def make_h5(image_dir, h5_path, image_names = [], indx_train = None, indx_valid = None, image_labels = None, image_types = None):
-
-    N = len(image_names)
-
-    # Process none
-    if indx_train is None:
-        indx_train = np.array(range(1, N, 2), dtype=np.int32)
-    if indx_valid is None:
-        indx_valid = np.array(range(0, N, 2), dtype=np.int32)
-    if image_labels is None:
-        image_labels = np.zeros(N)
-    if image_types is None:
-        image_types = np.zeros(N)
-
-    # Get image data
-    image_datas = []
-    for i in range(N):
-        image_name = image_names[i]
-        image_path = os.path.join(image_dir, image_name)
-        image = Image.open(image_path).convert("RGB")
-        rgb_array = np.array(image).transpose((2, 0, 1))
-        image_datas.append(rgb_array)
-    image_datas = np.array(image_datas, dtype=np.uint8)
-
-    # Create a new HDF5 file
-    with h5py.File(h5_path, "w") as f:
-        f.create_dataset("images", data=image_datas)
-        f.create_dataset("indx_train", data=indx_train)
-        f.create_dataset("indx_valid", data=indx_valid)
-        f.create_dataset("labels", data=image_labels)
-        f.create_dataset("types", data=image_types)
-
-        # Visualize
-        f.visititems(print_hdf5)
-
-__all__ = ["make_h5"]
+
+datasets_dir = resources.files("myosotis_researches").joinpath("CcGAN", "datasets")
+
+
+def make_h5(
+    old_datasets_name,
+    size,
+    new_datasets_path,
+    image_indexes,
+    train_indexes,
+    val_indexes,
+):
+
+    old_datasets_path = os.path.join(
+        datasets_dir, old_datasets_name, f"{old_datasets_name}_{size}x{size}.h5"
+    )
+
+    with h5py.File(old_datasets_path, "r") as f:
+        image_datas = f["images"]
+        image_labels = f["labels"]
+        image_types = f["types"]
+
+    with h5py.File(new_datasets_path, "w") as f:
+        f.create_dataset("images", data=image_datas[image_indexes])
+        f.create_dataset("indx_train", data=train_indexes)
+        f.create_dataset("indx_valid", data=val_indexes)
+        f.create_dataset("labels", data=image_labels[image_indexes])
+        f.create_dataset("types", data=image_types[image_indexes])
+
+
+__all__ = ["make_h5"]

{myosotis_researches-0.1.0.dist-info → myosotis_researches-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.1.0
+Version: 0.1.2
 Summary: A repository for storing my progress of researches.
 Home-page: https://github.com/Zeyu-Xie/Myosotis-Researches
 Author: Zeyu Xie
@@ -59,10 +59,11 @@ Import with code
 from myosotis_researches.utils import *
 ```
 
-| Function                                              | Description                               |
-| ----------------------------------------------------- | ----------------------------------------- |
-| `concat_image(img_list, gap=2, direction="vertical")` | Concat images vertically or horizontally. |
-| `print_hdf5(name, obj)`                               | Print a basic structure of an HDF5 file.  |
+| Function                                                     | Description                               |
+| ------------------------------------------------------------ | ----------------------------------------- |
+| `concat_image(img_list, gap=2, direction="vertical")`        | Concat images vertically or horizontally. |
+| `make_h5(old_datasets_name, size, new_datasets_path, image_indexes, train_indexes, val_indexes)` | Get piece of original HDF5 datasets.      |
+| `print_hdf5(name, obj)`                                      | Print a basic structure of an HDF5 file.  |
 
 **Note**:
 

{myosotis_researches-0.1.0.dist-info → myosotis_researches-0.1.2.dist-info}/RECORD

@@ -45,11 +45,11 @@ myosotis_researches/CcGAN/train_128_output_10/train_cgan_concat.py,sha256=PYctY3
 myosotis_researches/CcGAN/train_128_output_10/train_net_for_label_embed.py,sha256=4j6r4_o4rXgAN4MdUQL-TXqZJpbhH7d9gWQR8YzBlXw,6976
 myosotis_researches/CcGAN/train_128_output_10/utils.py,sha256=B-V6ct4WDisVVCOLO0W7VIBL8StPVNJJTZZ2b2NkMFU,3766
 myosotis_researches/CcGAN/utils/__init__.py,sha256=azZ2ZSSmWREoptI_5oQ180HojMoCqv2oleveRswq40w,155
-myosotis_researches/CcGAN/utils/concat_image.py,sha256=zSVFkIgj18m9wKTNWxBmZIdqMgVKF2IB2m1C65l_qyo,2151
-myosotis_researches/CcGAN/utils/make_h5.py,sha256=Jzsr9q7lneHyvi5HX1m4WBNarUYU1jUsXJuHnyQlAGc,1392
+myosotis_researches/CcGAN/utils/concat_image.py,sha256=BIGKz52Inn9S7M5fBFKye2V9bLJ0DqEQILoOVWAXUiE,2165
+myosotis_researches/CcGAN/utils/make_h5.py,sha256=HS0wzFVMh_Hh2xxoOpc47SUfhU6S2wR0wtveDJUkpOg,940
 myosotis_researches/CcGAN/utils/print_hdf5.py,sha256=VvmNAWtMDmg6D9V6ZbSUXrQTKRh9WIJeC4BR_ORJkco,300
-myosotis_researches-0.1.0.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.1.0.dist-info/METADATA,sha256=B2lXNQkCL9ds_JlE0M2kbjMSQZlSNT7tFUxO4_zEcoU,2490
-myosotis_researches-0.1.0.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-myosotis_researches-0.1.0.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.1.0.dist-info/RECORD,,
+myosotis_researches-0.1.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+myosotis_researches-0.1.2.dist-info/METADATA,sha256=uDjW4mHEPM12GtiuomEAPnk_GJbBD9Pwxwo3_lBZ4G8,2663
+myosotis_researches-0.1.2.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+myosotis_researches-0.1.2.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
+myosotis_researches-0.1.2.dist-info/RECORD,,