Myosotis-Researches 0.0.12__py3-none-any.whl → 0.0.13__py3-none-any.whl

myosotis_researches/CcGAN/utils/make_h5.py

@@ -5,10 +5,20 @@ 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):
+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 == None:
+        indx_train = np.array(range(1, N, 2), dtype=np.int32)
+    if indx_valid == None:
+        indx_valid = np.array(range(0, N, 2), dtype=np.int32)
+    if image_labels == None:
+        image_labels = np.zeros(N)
+    if image_types == None:
+        image_types = np.zeros(N)
+
     # Get image data
     image_datas = []
     for i in range(N):
@@ -19,17 +29,11 @@ def make_h5(image_dir: str, h5_path: str, image_names: list[str], image_labels,
         image_datas.append(rgb_array)
     image_datas = np.array(image_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("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)
 

{myosotis_researches-0.0.12.dist-info → myosotis_researches-0.0.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Myosotis-Researches
-Version: 0.0.12
+Version: 0.0.13
 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.12.dist-info → myosotis_researches-0.0.13.dist-info}/RECORD

@@ -3,10 +3,10 @@ myosotis_researches/CcGAN/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 myosotis_researches/CcGAN/utils/__init__.py,sha256=Pu9COV4zcXHGXuczhObersyeshVChmlEtwqp8VLUDxw,300
 myosotis_researches/CcGAN/utils/concat_image_horizontal.py,sha256=e6WsfO9IiSoP8zkZNz7IGimPUASr9VvyJUJdF-d40iw,954
 myosotis_researches/CcGAN/utils/concat_image_vertical.py,sha256=97-SuE8ZWpaeBm_ed6MAEaUOvtpzlYq_X3yWt4OEUTY,951
-myosotis_researches/CcGAN/utils/make_h5.py,sha256=baY4lElNUzoCkEcoYrE1bulEYFHr1l6vKAof2WtbMQI,1239
+myosotis_researches/CcGAN/utils/make_h5.py,sha256=bZGNx_SgWOaG8h3FpRk4WffC70hF6f9iIwj8z6dCKHI,1421
 myosotis_researches/CcGAN/utils/print_hdf5_structure.py,sha256=leaR8H3GhlX6EuIXDMh36xG2zBdV-XlJkaXBuoorl6I,320
-myosotis_researches-0.0.12.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.0.12.dist-info/METADATA,sha256=_awkE9siKuYkbBqvK9sc9p69z7KN6NJh8Pxf3Qqp994,765
-myosotis_researches-0.0.12.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-myosotis_researches-0.0.12.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.0.12.dist-info/RECORD,,
+myosotis_researches-0.0.13.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+myosotis_researches-0.0.13.dist-info/METADATA,sha256=2fWviTcvbMENj0D2O6asJIdg5wBWIGJLTqZgo3juQlk,765
+myosotis_researches-0.0.13.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+myosotis_researches-0.0.13.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
+myosotis_researches-0.0.13.dist-info/RECORD,,