Myosotis-Researches 0.1.10__tar.gz → 0.1.11__tar.gz

myosotis_researches-0.1.11/MANIFEST.in

@@ -0,0 +1 @@
+include myosotis_researches/CcGAN/visualize/src/

{myosotis_researches-0.1.10 → myosotis_researches-0.1.11/Myosotis_Researches.egg-info}/PKG-INFO

@@ -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 → myosotis_researches-0.1.11}/Myosotis_Researches.egg-info/SOURCES.txt

@@ -1,4 +1,5 @@
 LICENSE
+MANIFEST.in
 README.md
 pyproject.toml
 setup.py
@@ -40,4 +41,6 @@ myosotis_researches/CcGAN/utils/concat_image.py
 myosotis_researches/CcGAN/utils/make_h5.py
 myosotis_researches/CcGAN/utils/opts.py
 myosotis_researches/CcGAN/utils/print_hdf5.py
-myosotis_researches/CcGAN/utils/train.py
+myosotis_researches/CcGAN/utils/train.py
+myosotis_researches/CcGAN/visualize/__init__.py
+myosotis_researches/CcGAN/visualize/visualize_datasets.py

{myosotis_researches-0.1.10/Myosotis_Researches.egg-info → myosotis_researches-0.1.11}/PKG-INFO

@@ -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.11/myosotis_researches/CcGAN/visualize/__init__.py

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

myosotis_researches-0.1.11/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 → myosotis_researches-0.1.11}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="Myosotis-Researches",
-    version="0.1.10",
+    version="0.1.11",
     description="A repository for storing my progress of researches.",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",
@@ -10,6 +10,7 @@ setup(
     author_email="xie.zeyu20@gmail.com",
     url="https://github.com/Zeyu-Xie/Myosotis-Researches",
     packages=find_packages(),
+    include_package_data=True,
     install_requires=[],
     classifiers=[
         "Programming Language :: Python :: 3",