prefab 0.4.6__py3-none-any.whl → 0.4.7__py3-none-any.whl

prefab/predictor.py

@@ -4,9 +4,11 @@ using machine learning models deployed in the cloud.
 """
 
 import base64
+
 import numpy as np
 import requests
-from cv2 import imencode, imdecode, IMREAD_GRAYSCALE
+from cv2 import IMREAD_GRAYSCALE, imdecode, imencode
+
 from prefab.processor import binarize_hard
 
 
@@ -49,6 +51,7 @@ def predict(
         "device": _encode_image(device),
         "model_name": model_name,
         "model_tags": model_tags,
+        "binary": binarize,
     }
 
     prediction = _decode_image(
@@ -62,7 +65,11 @@ def predict(
 
 
 def correct(
-    device: np.ndarray, model_name: str, model_tags: str, binarize: bool = False
+    device: np.ndarray,
+    model_name: str,
+    model_tags: str,
+    binarize: bool = True,
+    multi_correct: bool = False,
 ) -> np.ndarray:
     """
     Generates a correction for a photonic device using a specified cloud-based ML model.
@@ -84,7 +91,11 @@ def correct(
         Consult the module's documentation for available tags.
 
     binarize : bool, optional
-        If set to True, the correction will be binarized (default is False).
+        If set to True, the correction will be binarized (default is True).
+
+    multi_correct : bool, optional
+        If set to True, the correction will be generated using a iterative approach.
+        (default is False).
 
     Returns
     -------
@@ -100,15 +111,14 @@ def correct(
         "device": _encode_image(device),
         "model_name": model_name,
         "model_tags": model_tags,
+        "binary": binarize,
+        "multi_correct": multi_correct,
     }
 
     correction = _decode_image(
         requests.post(function_url, json=correct_data, timeout=200)
     )
 
-    if binarize:
-        correction = binarize_hard(correction)
-
     return correction
 
 

{prefab-0.4.6.dist-info → prefab-0.4.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prefab
-Version: 0.4.6
+Version: 0.4.7
 Summary: Machine learning based prediction of photonic device fabrication
 Project-URL: Homepage, https://github.com/PreFab-Photonics/PreFab
 Author-email: Dusan Gostimirovic <dusan@prefabphotonics.com>
@@ -545,7 +545,7 @@ Description-Content-Type: text/markdown
 
 | Foundry   | Process                                                                            | Latest Version  | Latest Dataset   | Model Name    | Model Tag   | Status |
 | --------- | ---------------------------------------------------------------------------------- | --------------- | ---------------- | ------------- | ----------- | ------ |
-| ANT       | [NanoSOI](https://www.appliednt.com/nanosoi-fabrication-service/)                  | v5 (Jun 3 2023) | d4 (Apr 12 2023) | ANT_NanoSOI   | v5-d4       | Beta   |
+| ANT       | [NanoSOI](https://www.appliednt.com/nanosoi-fabrication-service/)                  | v6 (Nov 24 2023) | d7 (Nov 24 2023) | ANT_NanoSOI   | v6-d7       | Beta   |
 | ANT       | [SiN (Upper Edge)](https://www.appliednt.com/nanosoi/sys/resources/specs_nitride/) | v5 (Jun 3 2023) | d0 (Jun 1 2023)  | ANT_SiN       | v5-d0-upper | Alpha  |
 | ANT       | [SiN (Lower Edge)](https://www.appliednt.com/nanosoi/sys/resources/specs_nitride/) | v5 (Jun 3 2023) | d0 (Jun 1 2023)  | ANT_SiN       | v5-d0-lower | Alpha  |
 | SiEPICfab | [SOI](https://siepic.ca/fabrication/)                                              | v5 (Jun 3 2023) | d0 (Jun 14 2023) | SiEPICfab_SOI | v5-d0       | Alpha  |

prefab-0.4.7.dist-info/RECORD

@@ -0,0 +1,8 @@
+prefab/__init__.py,sha256=0r0FhWL0XnAE_rq9OuqrNsXNOWjU8eI3-wMvHVDBHWc,1461
+prefab/io.py,sha256=6UBP3KHTLpmcGkShH6ip9IU1RPCoOp5n8kEjeOjBgfk,6024
+prefab/predictor.py,sha256=UnftijiEUlAPEn-kiofD5Fmi3GmwgO5r-ku7JLBH_Y0,4788
+prefab/processor.py,sha256=lmSOTouyFOb8UZo7JLXi78VjdUz6eJoOyPZMn6IKraw,8996
+prefab-0.4.7.dist-info/METADATA,sha256=a9AzQpyVCGOl-fPFq_85e4kmzDKR915_Dh2WQ4tTo4o,35015
+prefab-0.4.7.dist-info/WHEEL,sha256=mRYSEL3Ih6g5a_CVMIcwiF__0Ae4_gLYh01YFNwiq1k,87
+prefab-0.4.7.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+prefab-0.4.7.dist-info/RECORD,,

prefab-0.4.6.dist-info/RECORD

@@ -1,8 +0,0 @@
-prefab/__init__.py,sha256=0r0FhWL0XnAE_rq9OuqrNsXNOWjU8eI3-wMvHVDBHWc,1461
-prefab/io.py,sha256=6UBP3KHTLpmcGkShH6ip9IU1RPCoOp5n8kEjeOjBgfk,6024
-prefab/predictor.py,sha256=BvcmoUuN3g_MNb9IichHMmueqQILnQK7U8LureMXJec,4562
-prefab/processor.py,sha256=lmSOTouyFOb8UZo7JLXi78VjdUz6eJoOyPZMn6IKraw,8996
-prefab-0.4.6.dist-info/METADATA,sha256=y4BRtxeYUaPIOy2oNHT_G65WuM8PoZKcy5Y_5xgsNBU,35014
-prefab-0.4.6.dist-info/WHEEL,sha256=mRYSEL3Ih6g5a_CVMIcwiF__0Ae4_gLYh01YFNwiq1k,87
-prefab-0.4.6.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-prefab-0.4.6.dist-info/RECORD,,