geococo.coco_models

Classes

CocoDataset

Info

Image

Annotation

Category

RleDict

dict() -> new empty dictionary

Source

Module Contents

class geococo.coco_models.CocoDataset[source]

Bases: pydantic.BaseModel

info: Info
images: List[Image] = []
annotations: List[Annotation] = []
categories: List[Category] = []
sources: List[Source] = []
next_image_id: int
next_annotation_id: int
next_source_id: int
_set_ids(values)[source]
Parameters:

values (Dict[str, Any])

Return type:

Dict[str, Any]

add_annotation(annotation)[source]
Parameters:

annotation (Annotation)

Return type:

None

add_image(image)[source]
Parameters:

image (Image)

Return type:

None

add_source(source_path, date_captured)[source]
Parameters:
Return type:

None

add_categories(category_ids, category_names, super_names)[source]
Parameters:

  • category_ids (Optional[pandas.Series])

  • category_names (Optional[pandas.Series])

  • super_names (Optional[pandas.Series])

Return type:

None

bump_version(bump_method)[source]
Parameters:

bump_method (str)

Return type:

None

verify_used_dir(images_dir)[source]
Parameters:

images_dir (pathlib.Path)

Return type:

None

class geococo.coco_models.Info[source]

Bases: pydantic.BaseModel

version: str = ''
year: int | None = None
description: str | None = None
contributor: str | None = None
date_created: datetime.datetime | None = None
class geococo.coco_models.Image[source]

Bases: pydantic.BaseModel

id: int
width: int
height: int
file_name: pathlib.Path
source_id: int
date_captured: datetime.datetime
class geococo.coco_models.Annotation[source]

Bases: pydantic.BaseModel

id: int
image_id: int
category_id: int
segmentation: RleDict
area: float
bbox: List[int]
iscrowd: int
class geococo.coco_models.Category[source]

Bases: pydantic.BaseModel

id: int
name: str
supercategory: str
class geococo.coco_models.RleDict[source]

Bases: typing_extensions.TypedDict

dict() -> new empty dictionary dict(mapping) -> new dictionary initialized from a mapping object’s

(key, value) pairs

dict(iterable) -> new dictionary initialized as if via:

d = {} for k, v in iterable:

d[k] = v

dict(**kwargs) -> new dictionary initialized with the name=value pairs

in the keyword argument list. For example: dict(one=1, two=2)

size: List[int]
counts: bytes
class geococo.coco_models.Source[source]

Bases: pydantic.BaseModel

id: int
file_name: pathlib.Path
date_captured: datetime.datetime