code-loader 1.0.38.dev6__py3-none-any.whl → 1.0.40__py3-none-any.whl

code_loader/contract/responsedataclasses.py

@@ -1,4 +1,4 @@
-from typing import List, Optional, Dict, Any
+from typing import List, Optional, Dict, Any, Union
 
 from dataclasses import dataclass, field
 from code_loader.contract.enums import DatasetMetadataType, LeapDataType
@@ -104,6 +104,7 @@ class BoundingBox:
     confidence: float
     label: str
     rotation: float = 0.0  # value between [0, 360], represent the degree of rotation.
+    metadata: Optional[Dict[str, Union[str, int, float]]] = None
 
 
 @dataclass

code_loader/contract/visualizer_classes.py

@@ -151,7 +151,10 @@ class LeapImageWithHeatmap:
         validate_type(type(self.labels), list)
         for label in self.labels:
             validate_type(type(label), str)
-            
+        if self.heatmaps.shape[0] != len(self.labels):
+            raise LeapValidationError(
+                'Number of heatmaps and labels must be equal')
+
 
 map_leap_data_type_to_visualizer_class = {
     LeapDataType.Image.value: LeapImage,

code_loader/inner_leap_binder/leapbinder.py

@@ -17,7 +17,7 @@ from code_loader.utils import to_numpy_return_wrapper, get_shape
 from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
     default_graph_visualizer, \
     default_image_visualizer, default_horizontal_bar_visualizer, default_word_visualizer, \
-    default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer, default_image_heatmap_visualizer
+    default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer
 
 
 class LeapBinder:
@@ -31,8 +31,6 @@ class LeapBinder:
     def _extend_with_default_visualizers(self) -> None:
         self.set_visualizer(function=default_image_visualizer, name=DefaultVisualizer.Image.value,
                             visualizer_type=LeapDataType.Image)
-        self.set_visualizer(function=default_image_heatmap_visualizer, name=DefaultVisualizer.ImageHeatmap.value,
-                            visualizer_type=LeapDataType.ImageWithHeatmap)
         self.set_visualizer(function=default_graph_visualizer, name=DefaultVisualizer.Graph.value,
                             visualizer_type=LeapDataType.Graph)
         self.set_visualizer(function=default_raw_data_visualizer, name=DefaultVisualizer.RawData.value,

code_loader/visualizers/default_visualizers.py

@@ -1,10 +1,11 @@
 from enum import Enum
+from typing import List
 
 import numpy as np
 import numpy.typing as npt
 
 from code_loader.contract.visualizer_classes import LeapImage, LeapGraph, LeapHorizontalBar, LeapText, \
-    LeapImageMask, LeapTextMask, LeapImageWithHeatmap
+    LeapImageMask, LeapTextMask
 from code_loader.utils import rescale_min_max
 
 
@@ -16,7 +17,6 @@ class DefaultVisualizer(Enum):
     ImageMask = 'ImageMask'
     TextMask = 'TextMask'
     RawData = 'RawData'
-    ImageHeatmap = 'ImageHeatmap'
     
 
 def default_image_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
@@ -24,15 +24,6 @@ def default_image_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
     return LeapImage(rescaled_data)
 
 
-def default_image_heatmap_visualizer(image: npt.NDArray[np.float32], heatmaps: npt.NDArray[np.float32], labels: list) -> LeapImageWithHeatmap:
-    rescaled_image = rescale_min_max(image)
-    rescaled_heatmaps = []
-    for heatmap in heatmaps:
-        rescaled_heatmap = rescale_min_max(heatmap)
-        rescaled_heatmaps = np.append(rescaled_heatmaps, rescaled_heatmap)
-    return LeapImageWithHeatmap(rescaled_image, rescaled_heatmaps, labels)
-
-
 def default_graph_visualizer(data: npt.NDArray[np.float32]) -> LeapGraph:
     return LeapGraph(data)
 

{code_loader-1.0.38.dev6.dist-info → code_loader-1.0.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.38.dev6
+Version: 1.0.40
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.38.dev6.dist-info → code_loader-1.0.40.dist-info}/RECORD

@@ -4,15 +4,15 @@ code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 code_loader/contract/datasetclasses.py,sha256=e4pEajPiktogtrctf29Q9IqG5VaZz0QE_ioc4Ri1ohs,4796
 code_loader/contract/enums.py,sha256=6Lo7p5CUog68Fd31bCozIuOgIp_IhSiPqWWph2k3OGU,1602
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
-code_loader/contract/responsedataclasses.py,sha256=WSHmFZWOFhGL1eED1u-aoRotPQg2owFQ-t3xSViWXSI,2808
-code_loader/contract/visualizer_classes.py,sha256=fP7lzIq1SnvxZhfUpC4Q86yxUTaEllYu4pRJ5HaaJdU,5845
+code_loader/contract/responsedataclasses.py,sha256=JgISOk1fVAD7dBLmwp2IzCkpItwO7YP3SDqzTX5FMDA,2880
+code_loader/contract/visualizer_classes.py,sha256=EKkQ7qWS48LeT3JoJT8-uMdRDxVOkSp0q-rhVMxrEFY,5990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
-code_loader/inner_leap_binder/leapbinder.py,sha256=KpTAkzP9JCSW6fIWKT1ZRtbmSLcC629UoSWSpt88uQs,13833
+code_loader/inner_leap_binder/leapbinder.py,sha256=cayCJxnksr5lrEIX1HngiuPpTupStti4DD6KX3k-PBI,13610
 code_loader/leaploader.py,sha256=pUySweZetJ6SsubCcZlDCJpvWmUrm5YlPlkZWQxY1hQ,17289
 code_loader/utils.py,sha256=61I4PgSl-ZBIe4DifLxMNlBELE-HQR2pB9efVYPceIU,2230
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/visualizers/default_visualizers.py,sha256=mts54M0JuMkNvBklv_PVqobmj1gcEEO44LxiFAvRivI,2733
-code_loader-1.0.38.dev6.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.38.dev6.dist-info/METADATA,sha256=x9KacwbSL-1MSrc8n6cXSFUTyeeY5r_fJzOmaoNNawI,773
-code_loader-1.0.38.dev6.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-code_loader-1.0.38.dev6.dist-info/RECORD,,
+code_loader/visualizers/default_visualizers.py,sha256=F2qRT6rNy7SOmr4QfqVxIkLlYEa00CwDkLJuA45lfSI,2254
+code_loader-1.0.40.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.40.dist-info/METADATA,sha256=9mNqefJJnuDJtTUEAfuen4-4m44xS2BmiqGP-uPS4PI,768
+code_loader-1.0.40.dist-info/WHEEL,sha256=kLuE8m1WYU0Ig0_YEGrXyTtiJvKPpLpDEiChiNyei5Y,88
+code_loader-1.0.40.dist-info/RECORD,,

{code_loader-1.0.38.dev6.dist-info → code_loader-1.0.40.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.8.1
+Generator: poetry-core 1.5.1
 Root-Is-Purelib: true
 Tag: py3-none-any