code-loader 1.0.81a0__py3-none-any.whl → 1.0.83__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 (6) hide show
  1. code_loader/contract/enums.py +1 -0
  2. code_loader/contract/visualizer_classes.py +25 -0
  3. {code_loader-1.0.81a0.dist-info → code_loader-1.0.83.dist-info}/METADATA +1 -1
  4. {code_loader-1.0.81a0.dist-info → code_loader-1.0.83.dist-info}/RECORD +6 -6
  5. {code_loader-1.0.81a0.dist-info → code_loader-1.0.83.dist-info}/LICENSE +0 -0
  6. {code_loader-1.0.81a0.dist-info → code_loader-1.0.83.dist-info}/WHEEL +0 -0
code_loader/contract/enums.py CHANGED
@@ -27,6 +27,7 @@ class LeapDataType(Enum):
27
27
  TextMask = 'TextMask'
28
28
  ImageWithBBox = 'ImageWithBBox'
29
29
  ImageWithHeatmap = 'ImageWithHeatmap'
30
+ Video = 'Video'
30
31
 
31
32
 
32
33
  class MetricDirection(Enum):
code_loader/contract/visualizer_classes.py CHANGED
@@ -53,6 +53,30 @@ class LeapImage:
53
53
  validate_type(type(self.compress), bool, 'compress flag must be a boolean')
54
54
 
55
55
 
56
+ @dataclass
57
+ class LeapVideo:
58
+ """
59
+ Visualizer representing a video for Tensorleap.
60
+
61
+ Attributes:
62
+ data (npt.NDArray[np.float32] | npt.NDArray[np.uint8]): The video data, shaped [T, H, W, C], where T is the number of frames.
63
+ type (LeapDataType): The data type, default is LeapDataType.Video.
64
+
65
+ Example:
66
+ video_data = np.random.rand(10, 100, 100, 3).astype(np.float32)
67
+ leap_video = LeapVideo(data=video_data)
68
+ """
69
+ data: Union[npt.NDArray[np.float32], npt.NDArray[np.uint8]]
70
+ type: LeapDataType = LeapDataType.Video
71
+
72
+ def __post_init__(self) -> None:
73
+ validate_type(self.type, LeapDataType.Video)
74
+ validate_type(type(self.data), np.ndarray)
75
+ validate_type(self.data.dtype, [np.uint8, np.float32])
76
+ validate_type(len(self.data.shape), 4, 'Video must be of shape 4')
77
+ validate_type(self.data.shape[3], [1, 3], 'Video channel must be either 3(rgb) or 1(gray)')
78
+
79
+
56
80
  @dataclass
57
81
  class LeapImageWithBBox:
58
82
  """
@@ -308,6 +332,7 @@ class LeapImageWithHeatmap:
308
332
  map_leap_data_type_to_visualizer_class = {
309
333
  LeapDataType.Image.value: LeapImage,
310
334
  LeapDataType.Graph.value: LeapGraph,
335
+ LeapDataType.Video.value: LeapVideo,
311
336
  LeapDataType.Text.value: LeapText,
312
337
  LeapDataType.HorizontalBar.value: LeapHorizontalBar,
313
338
  LeapDataType.ImageMask.value: LeapImageMask,
{code_loader-1.0.81a0.dist-info → code_loader-1.0.83.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 1.0.81a0
3
+ Version: 1.0.83
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
{code_loader-1.0.81a0.dist-info → code_loader-1.0.83.dist-info}/RECORD RENAMED
@@ -2,10 +2,10 @@ LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
2
2
  code_loader/__init__.py,sha256=6MMWr0ObOU7hkqQKgOqp4Zp3I28L7joGC9iCbQYtAJg,241
3
3
  code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  code_loader/contract/datasetclasses.py,sha256=rpsccWddYksLQhB0fDZKZsiuclb32wtAraLtvXZ0teM,7912
5
- code_loader/contract/enums.py,sha256=6Lo7p5CUog68Fd31bCozIuOgIp_IhSiPqWWph2k3OGU,1602
5
+ code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
6
6
  code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
7
7
  code_loader/contract/responsedataclasses.py,sha256=RSx9m_R3LawhK5o1nAcO3hfp2F9oJYtxZr_bpP3bTmw,4005
8
- code_loader/contract/visualizer_classes.py,sha256=m31lg2P2QJs3Reqr6-N1AlVhH3RxPr772Jw3LuIVCVM,14177
8
+ code_loader/contract/visualizer_classes.py,sha256=_nlukRfW8QeQaQG7G5HfAUSeCuoqslB4xJS2AGI_Nz8,15156
9
9
  code_loader/default_losses.py,sha256=NoOQym1106bDN5dcIk56Elr7ZG5quUHArqfP5-Nyxyo,1139
10
10
  code_loader/default_metrics.py,sha256=v16Mrt2Ze1tXPgfKywGVdRSrkaK4CKLNQztN1UdVqIY,5010
11
11
  code_loader/experiment_api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,7 +26,7 @@ code_loader/leaploaderbase.py,sha256=VH0vddRmkqLtcDlYPCO7hfz1_VbKo43lUdHDAbd4iJc
26
26
  code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
27
27
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
28
28
  code_loader/visualizers/default_visualizers.py,sha256=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
29
- code_loader-1.0.81a0.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
30
- code_loader-1.0.81a0.dist-info/METADATA,sha256=g0WzhZyKvBxa0ufE_U0ujUNR1PITvSmAblWtmIDglsc,851
31
- code_loader-1.0.81a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
32
- code_loader-1.0.81a0.dist-info/RECORD,,
29
+ code_loader-1.0.83.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
30
+ code_loader-1.0.83.dist-info/METADATA,sha256=tswBdvB4I6DTyR_sZaGqwgInOTwo8RFLZo3GgTqCsmE,849
31
+ code_loader-1.0.83.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
32
+ code_loader-1.0.83.dist-info/RECORD,,