Skip to content

Commit cf25f3a

Browse files
Vb/remove dicomm plt 1853 (#1902)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
1 parent ef10587 commit cf25f3a

File tree

10 files changed

+55
-875
lines changed

10 files changed

+55
-875
lines changed

examples/README.md

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -143,11 +143,6 @@
143143
</tr>
144144
</thead>
145145
<tbody>
146-
<tr>
147-
<td>DICOM</td>
148-
<td><a href="https://github.com/Labelbox/labelbox-python/tree/develop/examples/annotation_import/dicom.ipynb" target="_blank"><img src="https://img.shields.io/badge/GitHub-100000?logo=github&logoColor=white" alt="Open In Github"></a></td>
149-
<td><a href="https://colab.research.google.com/github/Labelbox/labelbox-python/blob/develop/examples/annotation_import/dicom.ipynb" target="_blank"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a></td>
150-
</tr>
151146
<tr>
152147
<td>Tiled</td>
153148
<td><a href="https://github.com/Labelbox/labelbox-python/tree/develop/examples/annotation_import/tiled.ipynb" target="_blank"><img src="https://img.shields.io/badge/GitHub-100000?logo=github&logoColor=white" alt="Open In Github"></a></td>

examples/annotation_import/dicom.ipynb

Lines changed: 0 additions & 266 deletions
This file was deleted.

libs/labelbox/src/labelbox/data/annotation_types/__init__.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,9 @@
1515

1616
from .video import VideoClassificationAnnotation
1717
from .video import VideoObjectAnnotation
18-
from .video import DICOMObjectAnnotation
19-
from .video import GroupKey
2018
from .video import MaskFrame
2119
from .video import MaskInstance
2220
from .video import VideoMaskAnnotation
23-
from .video import DICOMMaskAnnotation
2421

2522
from .ner import ConversationEntity
2623
from .ner import DocumentEntity

libs/labelbox/src/labelbox/data/annotation_types/video.py

Lines changed: 7 additions & 69 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,12 @@
1-
from enum import Enum
21
from typing import List, Optional, Tuple
32

4-
from labelbox.data.annotation_types.annotation import (
5-
ClassificationAnnotation,
6-
ObjectAnnotation,
3+
from pydantic import (
4+
AliasChoices,
5+
BaseModel,
6+
ConfigDict,
7+
Field,
8+
field_validator,
9+
model_validator,
710
)
811

912
from labelbox.data.annotation_types.annotation import (
@@ -16,14 +19,6 @@
1619
CustomMetricsNotSupportedMixin,
1720
)
1821
from labelbox.utils import _CamelCaseMixin, is_valid_uri
19-
from pydantic import (
20-
model_validator,
21-
BaseModel,
22-
field_validator,
23-
Field,
24-
ConfigDict,
25-
AliasChoices,
26-
)
2722

2823

2924
class VideoClassificationAnnotation(ClassificationAnnotation):
@@ -72,43 +67,6 @@ class VideoObjectAnnotation(
7267
segment_index: Optional[int] = None
7368

7469

75-
class GroupKey(Enum):
76-
"""Group key for DICOM annotations"""
77-
78-
AXIAL = "axial"
79-
SAGITTAL = "sagittal"
80-
CORONAL = "coronal"
81-
82-
83-
class DICOMObjectAnnotation(VideoObjectAnnotation):
84-
"""DICOM object annotation
85-
>>> DICOMObjectAnnotation(
86-
>>> name="dicom_polyline",
87-
>>> frame=2,
88-
>>> value=lb_types.Line(points = [
89-
>>> lb_types.Point(x=680, y=100),
90-
>>> lb_types.Point(x=100, y=190),
91-
>>> lb_types.Point(x=190, y=220)
92-
>>> ]),
93-
>>> segment_index=0,
94-
>>> keyframe=True,
95-
>>> Group_key=GroupKey.AXIAL
96-
>>> )
97-
Args:
98-
name (Optional[str])
99-
feature_schema_id (Optional[Cuid])
100-
value (Geometry)
101-
group_key (GroupKey)
102-
frame (Int): The frame index that this annotation corresponds to
103-
keyframe (bool): Whether or not this annotation was a human generated or interpolated annotation
104-
segment_id (Optional[Int]): Index of video segment this annotation belongs to
105-
classifications (List[ClassificationAnnotation]) = []
106-
extra (Dict[str, Any])
107-
"""
108-
109-
group_key: GroupKey
110-
111-
11270
class MaskFrame(_CamelCaseMixin, BaseModel):
11371
index: int
11472
instance_uri: Optional[str] = Field(
@@ -162,23 +120,3 @@ class VideoMaskAnnotation(BaseModel):
162120

163121
frames: List[MaskFrame]
164122
instances: List[MaskInstance]
165-
166-
167-
class DICOMMaskAnnotation(VideoMaskAnnotation):
168-
"""DICOM mask annotation
169-
>>> DICOMMaskAnnotation(
170-
>>> name="dicom_mask",
171-
>>> group_key=GroupKey.AXIAL,
172-
>>> frames=[
173-
>>> MaskFrame(index=1, instance_uri='https://storage.labelbox.com/cjhfn5y6s0pk507024nz1ocys%2F1d60856c-59b7-3060-2754-83f7e93e0d01-1?Expires=1666901963361&KeyName=labelbox-assets-key-3&Signature=t-2s2DB4YjFuWEFak0wxYqfBfZA'),
174-
>>> MaskFrame(index=5, instance_uri='https://storage.labelbox.com/cjhfn5y6s0pk507024nz1ocys1%2F1d60856c-59b7-3060-2754-83f7e93e0d01-1?Expires=1666901963361&KeyName=labelbox-assets-key-3&Signature=t-2s2DB4YjFuWEFak0wxYqfBfZA'),
175-
>>> ],
176-
>>> instances=[
177-
>>> MaskInstance(color_rgb=(0, 0, 255), name="mask1"),
178-
>>> MaskInstance(color_rgb=(0, 255, 0), name="mask2"),
179-
>>> MaskInstance(color_rgb=(255, 0, 0), name="mask3")
180-
>>> ]
181-
>>> )
182-
"""
183-
184-
group_key: GroupKey

0 commit comments

Comments
 (0)