Myosotis-Researches 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

myosotis_researches/CcGAN/visualize/__init__.py

@@ -0,0 +1,3 @@
+from .visualize_datasets import visualize_datasets
+
+__all__ = ["visualize_datasets"]

myosotis_researches/CcGAN/visualize/visualize_datasets.py

@@ -0,0 +1,98 @@
+import base64
+from flask import Flask, send_file
+import h5py
+from importlib import resources
+from io import BytesIO
+from jinja2 import Template
+from myosotis_researches.CcGAN.internal import *
+import numpy as np
+import pandas as pd
+from PIL import Image
+
+
+def Image_to_Base64(image: Image):
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return f"data:image/png;base64,{img_base64}"
+
+
+def datasets_to_html(indexes, datasets_path, list_path, template_path):
+
+    N = len(indexes)
+
+    # Get template
+    with open(template_path, "r") as f:
+        template = Template(f.read())
+
+    # Get data
+    with h5py.File(datasets_path, "r") as f:
+        images = f["images"][indexes]
+        index_train = f["index_train"][:]
+        index_valid = f["index_valid"][:]
+        labels = f["labels"][indexes]
+        types = f["types"][indexes]
+
+    # Get list
+    df = pd.read_csv(list_path)
+    tool_wears = df.loc[indexes, "tool_wear"]
+    surfaces = df.loc[indexes, "surface"]
+    image_names = df.loc[indexes, "image_name"]
+
+    # Transform data to base64
+    images = np.transpose(images, (0, 2, 3, 1))
+    images = [Image.fromarray(image) for image in images]
+    images_base64 = [Image_to_Base64(image) for image in images]
+
+    # Render template
+    items = []
+    for i in range(N):
+        items.append(
+            {
+                "image": images_base64[i],
+                "image_name": image_names[i],
+                "label": labels[i],
+                "type": types[i],
+                "index": indexes[i],
+            }
+        )
+
+    return template.render(
+        indexes=indexes,
+        datasets_path=datasets_path,
+        list_path=list_path,
+        template_path=template_path,
+        items=items,
+    )
+
+
+def visualize_datasets(
+    indexes,
+    datasets_path,
+    list_path,
+    template_path=resources.files("myosotis_researches").joinpath(
+        "CcGAN", "visualize", "src", "template.html"
+    ),
+    host="127.0.0.1",
+    port=8000,
+    debug=True,
+):
+    # Local server
+    app = Flask(__name__)
+
+    @app.route("/")
+    def index():
+        return datasets_to_html(indexes, datasets_path, list_path, template_path)
+
+    @app.route("/style.css")
+    def style():
+        return send_file(
+            resources.files("myosotis_researches").joinpath(
+                "CcGAN", "visualize", "src", "style.css"
+            )
+        )
+
+    app.run(host=host, port=port, debug=debug)
+
+
+__all__ = ["visualize_datasets"]

{myosotis_researches-0.1.10.dist-info → myosotis_researches-0.1.11.dist-info}/METADATA

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

@@ -33,8 +33,10 @@ myosotis_researches/CcGAN/utils/make_h5.py,sha256=VtFYjr_i-JktsEW_BvofpilcDmChRm
 myosotis_researches/CcGAN/utils/opts.py,sha256=pd7-wknNPBO5hWRpO3YAPmmAsPKgZUUpKc4gWMs6Wto,5397
 myosotis_researches/CcGAN/utils/print_hdf5.py,sha256=VvmNAWtMDmg6D9V6ZbSUXrQTKRh9WIJeC4BR_ORJkco,300
 myosotis_researches/CcGAN/utils/train.py,sha256=5ZXgkGesuInqUooJRpLej_KHqYQtlSDq90_5wig5elQ,5152
-myosotis_researches-0.1.10.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-myosotis_researches-0.1.10.dist-info/METADATA,sha256=tCHcXYDZ_af1keBiLUExIsWWoBIKxFdrY6wTWWm9L8c,2664
-myosotis_researches-0.1.10.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-myosotis_researches-0.1.10.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
-myosotis_researches-0.1.10.dist-info/RECORD,,
+myosotis_researches/CcGAN/visualize/__init__.py,sha256=RHfzFo7FgcJhZVQ9DkEIGELK-lQNC5X915kPH8XGSiM,85
+myosotis_researches/CcGAN/visualize/visualize_datasets.py,sha256=NcrszJon9mFtffMyIR8SJyxl-ZXWWEutFcViXzpn5e0,2533
+myosotis_researches-0.1.11.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+myosotis_researches-0.1.11.dist-info/METADATA,sha256=e7evcqF20wMz7S5Ls1oBSzK1uJ-FWAy47Auvix01MMc,2664
+myosotis_researches-0.1.11.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+myosotis_researches-0.1.11.dist-info/top_level.txt,sha256=zxAiMn5eyZNJM28MewTAkgi_RZJMbfWbzVR-KF0LdZE,20
+myosotis_researches-0.1.11.dist-info/RECORD,,