Coretex
coretex.coretex.annotation.image.classes_format.ImageDatasetClass Class Reference
Inheritance diagram for coretex.coretex.annotation.image.classes_format.ImageDatasetClass:
coretex.codable.codable.Codable coretex.coretex.annotation.image.classes_format.ImageDatasetClasses

Public Member Functions

ImageDatasetClasses generate (cls, Set[str] labels)
 
- Public Member Functions inherited from coretex.codable.codable.Codable
Dict[str, Any] encode (self)
 
None onDecode (self)
 
Self decode (cls, Dict[str, Any] encodedObject)
 

Detailed Description

    Image Dataset class metadata

    Properties
    ----------
    classIds : List[UUID]
        list of all uuids connected to this class
    label : str
        name of the class
    color : str
        color of the class

Definition at line 29 of file classes_format.py.

Member Function Documentation

◆ generate()

ImageDatasetClasses coretex.coretex.annotation.image.classes_format.ImageDatasetClass.generate (   cls,
Set[str]  labels 
)
    Generates list of Coretex classes based on provided
    labels (class names)

    Parameters
    ----------
    labels : Set[str]
list of labels (class names)

    Returns
    -------
    ImageDatasetClasses -> list of ImageDatasetClass (Coretex class metadata) objects

    Example
    -------
    >>> from coretex import ImageDatasetClass
    \b
    >>> labels = {"car", "bicycle"}
    >>> imgDatasetClasses = ImageDatasetClass.generate(labels)
    >>> print(classes)
    [
ImageDatasetClass(
    classIds = [UUID("81add001-1c9c-4949-8b73-3599f1d0de9d")],
    name = "car",
    color = "#bcb86b"
),
ImageDatasetClass(
    classIds = [UUID("d710019b-f28f-40ab-aa65-e13df949beff")],
    name = "bicycle",
    color = "#cbc66b"
)
    ]

Definition at line 69 of file classes_format.py.

69  def generate(cls, labels: Set[str]) -> ImageDatasetClasses:
70  """
71  Generates list of Coretex classes based on provided
72  labels (class names)
73 
74  Parameters
75  ----------
76  labels : Set[str]
77  list of labels (class names)
78 
79  Returns
80  -------
81  ImageDatasetClasses -> list of ImageDatasetClass (Coretex class metadata) objects
82 
83  Example
84  -------
85  >>> from coretex import ImageDatasetClass
86  \b
87  >>> labels = {"car", "bicycle"}
88  >>> imgDatasetClasses = ImageDatasetClass.generate(labels)
89  >>> print(classes)
90  [
91  ImageDatasetClass(
92  classIds = [UUID("81add001-1c9c-4949-8b73-3599f1d0de9d")],
93  name = "car",
94  color = "#bcb86b"
95  ),
96  ImageDatasetClass(
97  classIds = [UUID("d710019b-f28f-40ab-aa65-e13df949beff")],
98  name = "bicycle",
99  color = "#cbc66b"
100  )
101  ]
102  """
103 
104  colors: Set[str] = set()
105 
106  while len(colors) != len(labels):
107  color = f'#{"%06x" % random.randint(0, 0xFFFFFF)}'
108  colors.add(color)
109 
110  return ImageDatasetClasses(
111  [cls(label, color) for label, color in zip(labels, colors)]
112  )
113 
114 

The documentation for this class was generated from the following file: