Update annotations __init__

Author: kunlin596

dgp/annotations/__init__.py

@@ -20,6 +20,7 @@
 from dgp.annotations.semantic_segmentation_2d_annotation import SemanticSegmentation2DAnnotation  # isort:skip
 from dgp.annotations.key_line_2d_annotation import KeyLine2DAnnotationList  # isort:skip
 from dgp.annotations.key_line_3d_annotation import KeyLine3DAnnotationList  # isort:skip
+from dgp.annotations.key_line_3d_annotation import ProbabilisticKeyLine3DAnnotationList  # isort:skip
 from dgp.annotations.key_point_2d_annotation import KeyPoint2DAnnotationList  # isort:skip
 from dgp.annotations.key_point_3d_annotation import KeyPoint3DAnnotationList  # isort:skip
 from dgp.annotations.depth_annotation import DenseDepthAnnotation  # isort:skip
@@ -41,7 +42,7 @@
     "surface_normals_2d": None,
     "surface_normals_3d": None,
     "motion_vectors_2d": None,
-    "motion_vectors_3d": None
+    "motion_vectors_3d": None,
 }
 
 # Annotation objects for each annotation type
@@ -54,7 +55,8 @@
     "key_point_3d": KeyPoint3DAnnotationList,
     "key_line_2d": KeyLine2DAnnotationList,
     "key_line_3d": KeyLine3DAnnotationList,
-    "depth": DenseDepthAnnotation
+    "probabilistic_key_line_3d": ProbabilisticKeyLine3DAnnotationList,
+    "depth": DenseDepthAnnotation,
 }
 
 # Annotation groups for each annotation type: 2d/3d
@@ -73,5 +75,5 @@
     "key_line_2d": "2d",
     "key_point_3d": "3d",
     "key_line_3d": "3d",
-    "depth": "2d"
+    "depth": "2d",
 }