code-loader 1.0.89__py3-none-any.whl → 1.0.91__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (5) hide show
  1. code_loader/visualizers/default_visualizers.py +9 -4
  2. {code_loader-1.0.89.dist-info → code_loader-1.0.91.dist-info}/METADATA +1 -1
  3. {code_loader-1.0.89.dist-info → code_loader-1.0.91.dist-info}/RECORD +5 -5
  4. {code_loader-1.0.89.dist-info → code_loader-1.0.91.dist-info}/LICENSE +0 -0
  5. {code_loader-1.0.89.dist-info → code_loader-1.0.91.dist-info}/WHEEL +0 -0
code_loader/visualizers/default_visualizers.py CHANGED
@@ -20,21 +20,21 @@ class DefaultVisualizer(Enum):
20
20
 
21
21
 
22
22
  def default_image_visualizer(data: npt.NDArray[np.float32]) -> LeapImage:
23
- rescaled_data = rescale_min_max(data)
23
+ rescaled_data = rescale_min_max(data[0])
24
24
  return LeapImage(rescaled_data)
25
25
 
26
26
 
27
27
  def default_video_visualizer(data: npt.NDArray[np.float32]) -> LeapVideo:
28
- return LeapVideo(data)
28
+ return LeapVideo(data[0])
29
29
 
30
30
 
31
31
  def default_graph_visualizer(data: npt.NDArray[np.float32]) -> LeapGraph:
32
- return LeapGraph(data)
32
+ return LeapGraph(data[0])
33
33
 
34
34
 
35
35
  def default_horizontal_bar_visualizer(data: npt.NDArray[np.float32]) -> LeapHorizontalBar:
36
36
  labels = [str(index) for index in range(data.shape[-1])]
37
- return LeapHorizontalBar(data, labels)
37
+ return LeapHorizontalBar(data[0], labels)
38
38
 
39
39
 
40
40
  def default_word_visualizer(data: npt.NDArray[np.float32]) -> LeapText:
@@ -51,6 +51,11 @@ def default_raw_data_visualizer(data: npt.NDArray[np.float32]) -> LeapText:
51
51
 
52
52
 
53
53
  def default_image_mask_visualizer(mask: npt.NDArray[np.float32], image: npt.NDArray[np.float32]) -> LeapImageMask:
54
+ if image.shape[0] == 1:
55
+ image = image[0]
56
+ if mask.shape[0] == 1:
57
+ mask = mask[0]
58
+
54
59
  n_different_labels = mask.shape[-1]
55
60
  labels = [str(i) for i in range(n_different_labels)]
56
61
 
{code_loader-1.0.89.dist-info → code_loader-1.0.91.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 1.0.89
3
+ Version: 1.0.91
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
{code_loader-1.0.89.dist-info → code_loader-1.0.91.dist-info}/RECORD RENAMED
@@ -26,8 +26,8 @@ code_loader/leaploader.py,sha256=GGHZzovqGsM6oY14pwyL35LeOmdRfjVuhhY2ZqlYKaM,264
26
26
  code_loader/leaploaderbase.py,sha256=VH0vddRmkqLtcDlYPCO7hfz1_VbKo43lUdHDAbd4iJc,4198
27
27
  code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
28
28
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
- code_loader/visualizers/default_visualizers.py,sha256=f95WzLYO2KU8endWxrmtwjHx7diDfm8AaWjucYHT7GI,2439
30
- code_loader-1.0.89.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
31
- code_loader-1.0.89.dist-info/METADATA,sha256=RxcI-P4qzQtEJnG3r1wVOKAXEukB5gQYXc5opPe0e1U,849
32
- code_loader-1.0.89.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
33
- code_loader-1.0.89.dist-info/RECORD,,
29
+ code_loader/visualizers/default_visualizers.py,sha256=669lBpLISLO6my5Qcgn1FLDDeZgHumPf252m4KHY4YM,2555
30
+ code_loader-1.0.91.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
31
+ code_loader-1.0.91.dist-info/METADATA,sha256=xq7TJCYJ5JPkfAfRZOGpo_HZTIHx4XF7vGrXoXrh97s,849
32
+ code_loader-1.0.91.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
33
+ code_loader-1.0.91.dist-info/RECORD,,