Myosotis-Researches 0.0.7__tar.gz → 0.0.9__tar.gz

{myosotis_researches-0.0.7 → myosotis_researches-0.0.9/Myosotis_Researches.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.7
+Version: 0.0.9
 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.7 → myosotis_researches-0.0.9}/Myosotis_Researches.egg-info/SOURCES.txt

@@ -11,4 +11,5 @@ myosotis_researches/CcGAN/__init__.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/make_h5.py
 myosotis_researches/CcGAN/utils/print_hdf5_structure.py

{myosotis_researches-0.0.7/Myosotis_Researches.egg-info → myosotis_researches-0.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.7
+Version: 0.0.9
 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.7 → myosotis_researches-0.0.9}/myosotis_researches/CcGAN/utils/concat_image_horizontal.py

@@ -18,7 +18,7 @@ def concat_image_horizontal(img_list: list[Image], gap = 2) -> Image:
     
     # Create concated image
     concated_width = width * N + gap * (N+1)
-    concated_height = height * N + gap * 2
+    concated_height = height + gap * 2
     concated_image = Image.new("RGB", (concated_width, concated_height))
 
     # Copy ans paste

{myosotis_researches-0.0.7 → myosotis_researches-0.0.9}/myosotis_researches/CcGAN/utils/concat_image_vertical.py

@@ -17,7 +17,7 @@ def concat_image_vertical(img_list: list[Image], gap = 2) -> Image:
             raise ValueError("All images should have the same height.")
     
     # Create concated image
-    concated_width = width * N + gap * 2
+    concated_width = width + gap * 2
     concated_height = height * N + gap * (N+1)
     concated_image = Image.new("RGB", (concated_width, concated_height))
 

myosotis_researches-0.0.9/myosotis_researches/CcGAN/utils/make_h5.py

@@ -0,0 +1,39 @@
+import h5py
+import numpy as np
+import os
+from PIL import Image
+from print_hdf5_structure import print_hdf5_structure
+
+# Make all images to a HDF5 file
+def make_h5(image_dir: str, h5_path: str, image_names: list[str], image_labels, image_types):
+
+    N = len(image_names)
+
+    # 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)
+    img_datas = np.array(img_datas, dtype=np.uint8)
+
+    # Set train_idx = 1, 3, 5, ...
+    train_idx = np.array(range(1, N, 2), dtype=np.int32)
+
+    # Set val_idx = 0, 2, 4, ...
+    val_idx = np.array(range(0, N, 2), dtype=np.int32)
+
+    # 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=train_idx)
+        f.create_dataset("indx_valid", data=val_idx)
+        f.create_dataset("labels", data=image_labels)
+        f.create_dataset("types", data=image_types)
+
+        # Visualize
+        f.visititems(print_hdf5_structure)
+
+__all__ = ["make_h5"]

{myosotis_researches-0.0.7 → myosotis_researches-0.0.9}/setup.py

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