code-loader 1.0.211__tar.gz → 1.0.212__tar.gz

{code_loader-1.0.211 → code_loader-1.0.212}/PKG-INFO

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

{code_loader-1.0.211 → code_loader-1.0.212}/code_loader/contract/responsedataclasses.py

@@ -144,6 +144,7 @@ class BoundingBox3D:
     yaw (float): Rotation about +z in radians, counter-clockwise from +x. The +length/2 face is the front.
     label (str): The class name of the box.
     confidence (float): Prediction score, typically in [0, 1]; 1.0 for ground truth.
+    is_ground_truth (bool): True marks a ground-truth box; the UI draws it dashed so it can be shown with predictions.
     metadata (Optional[Dict[str, Union[str, int, float]]]): Optional extra fields shown with the box.
 
     Corners are the local points (±length/2, ±width/2, ±height/2) rotated by yaw about z, then translated to (x, y, z).
@@ -157,10 +158,10 @@ class BoundingBox3D:
     yaw: float
     label: str
     confidence: float = 1.0
+    is_ground_truth: bool = False
     metadata: Optional[Dict[str, Union[str, int, float]]] = None
 
 
-
 @dataclass
 class LeapAnalysisConfiguration:
     domain_gap_metadata: Optional[List[str]] = None

{code_loader-1.0.211 → code_loader-1.0.212}/code_loader/contract/visualizer_classes.py

@@ -395,8 +395,9 @@ class LeapPointCloud:
 
     Example:
         points = (np.random.rand(1000, 3) * 50).astype(np.float32)
-        box = BoundingBox3D(x=10, y=0, z=1, width=2, length=4.5, height=1.8, yaw=0.3, label="car", confidence=0.9)
-        leap_point_cloud = LeapPointCloud(points=points, intensity=points[:, 2].copy(), bounding_boxes=[box])
+        gt = BoundingBox3D(x=10, y=0, z=1, width=2, length=4.5, height=1.8, yaw=0.3, label="car", is_ground_truth=True)
+        pred = BoundingBox3D(x=10.2, y=0.1, z=1, width=2, length=4.4, height=1.8, yaw=0.28, label="car", confidence=0.9)
+        leap_point_cloud = LeapPointCloud(points=points, intensity=points[:, 2].copy(), bounding_boxes=[gt, pred])
     """
     points: npt.NDArray[np.float32]
     intensity: Optional[npt.NDArray[np.float32]] = None

{code_loader-1.0.211 → code_loader-1.0.212}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.211"
+version = "1.0.212"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"