VideoDataset¶

class VideoDataset[source]¶

Bases: supervisely.project.project.Dataset

VideoDataset is where your labeled and unlabeled videos and other data files live. VideoDataset object is immutable.

Parameters

directory : str: Path to dataset directory.
mode : OpenMode: Determines working mode for the given dataset.

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

Methods

`add_item_file`	Adds given item file to dataset items directory, and adds given annotation to dataset annotations directory.
`add_item_np`	Not available for VideoDataset class object.
`add_item_raw_bytes`	Not available for VideoDataset class object.
`delete_item`	Delete image, image info and annotation from `Dataset`.
`generate_item_path`	Generates full path to the given item.
`get_ann`	Read annotation of item from json.
`get_ann_path`	Path to the given annotation json file.
`get_classes_stats`
`get_image_info`	Not available for VideoDataset class object.
`get_img_info_path`	Not available for VideoDataset class object.
`get_img_path`	Not available for VideoDataset class object.
`get_item_info`	Information for Item with given name.
`get_item_info_path`	Get path to the item info json file without checking if the file exists.
`get_item_path`	Path to the given item.
`get_item_paths`	Generates `VideoItemPaths` object with paths to item and annotation directories for item with given name.
`get_items_names`	List of video dataset item names.
`get_seg_path`	Not available for VideoDataset class object.
`get_url`	Get URL to dataset items list in Supervisely.
`item_exists`	Checks if given item name belongs to the video dataset.
`items`	This method is used to iterate over dataset items, receiving item name, path to image and path to annotation json file.
`key`
`set_ann`	Replaces given annotation for given item name to dataset annotations directory in json format.
`set_ann_dict`	Replaces given annotation json for given item name to dataset annotations directory in json format.
`set_ann_file`	Replaces given annotation json file for given item name to dataset annotations directory in json format.

Attributes

`ann_dir`	Path to the video dataset annotations directory.
`ann_dir_name`	`str`: Annotations directory name
`directory`	Path to the video dataset directory.
`img_dir`	Not available for VideoDataset class object.
`img_info_dir`	Not available for VideoDataset class object.
`item_dir`	Path to the video dataset items directory.
`item_dir_name`	`str`: Items data directory name
`item_info_dir`	Path to the video dataset item with items info.
`item_info_dir_name`	`str`: Items info directory name
`name`	Video Dataset name.
`project_dir`	Path to the video project containing the video dataset.
`seg_dir`	Not available for VideoDataset class object.
`seg_dir_name`	`str`: Segmentation masks directory name

annotation_class¶: alias of supervisely.video_annotation.video_annotation.VideoAnnotation

item_info_class¶: alias of supervisely.api.video.video_api.VideoInfo

add_item_file(item_name, item_path, ann=None, _validate_item=True, _use_hardlink=False, item_info=None)[source]¶

Adds given item file to dataset items directory, and adds given annotation to dataset annotations directory. if ann is None, creates empty annotation file.

Parameters

item_name : str: Item name.
item_path : str: Path to the item.
ann : VideoAnnotation or str, optional: VideoAnnotation object or path to annotation json file.
_validate_item : bool, optional: Checks input files format.
_use_hardlink : bool, optional: If True creates a hardlink pointing to src named dst, otherwise don’t.
item_info : VideoInfo or dict or str, optional: VideoInfo object or VideoInfo object converted to dict or path to item info json file for copying to dataset item info directory.

Returns

None

Return type

NoneType

Raises

RuntimeError if item_name already exists in dataset or item name has unsupported extension.

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

ann = "/home/admin/work/supervisely/projects/videos_example/ds0/ann/video_8888.mp4.json"
ds.add_item_file("video_8888.mp4", "/home/admin/work/supervisely/projects/videos_example/ds0/video/video_8888.mp4", ann=ann)
print(ds.item_exists("video_8888.mp4"))
# Output: True

add_item_np(item_name, img, ann=None, img_info=None)[source]¶: Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

add_item_raw_bytes(item_name, item_raw_bytes, ann=None, img_info=None)[source]¶: Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

delete_item(item_name)¶

Delete image, image info and annotation from Dataset.

Parameters

item_name : str: Item name.

Returns

True if item was successfully deleted, False if item wasn’t found in dataset.

Return type

bool

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/lemons_annotated/ds1"
ds = sly.Dataset(dataset_path, sly.OpenMode.READ)

print(dataset.delete_item("IMG_0748"))
# Output: False

print(dataset.delete_item("IMG_0748.jpeg"))
# Output: True

generate_item_path(item_name)¶

Generates full path to the given item.

Parameters

item_name : str: Item name.

Returns

Full path to the given item

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/lemons_annotated/ds1"
ds = sly.Dataset(dataset_path, sly.OpenMode.READ)

print(ds.generate_item_path("IMG_0748.jpeg"))
# Output: '/home/admin/work/supervisely/projects/lemons_annotated/ds1/img/IMG_0748.jpeg'

get_ann(item_name, project_meta, key_id_map=None)[source]¶

Read annotation of item from json.

Parameters

item_name : str: Item name.
project_meta : ProjectMeta: ProjectMeta object.
key_id_map : KeyIdMap, optional: KeyIdMap object.

Returns

VideoAnnotation object.

Return type

VideoAnnotation

Raises

RuntimeError if item not found in the project

Usage example

import supervisely as sly
project_path = "/home/admin/work/supervisely/projects/videos_example"
project = sly.VideoProject(project_path, sly.OpenMode.READ)

ds = project.datasets.get('ds0')

annotation = ds.get_ann("video_0748", project.meta)
# Output: RuntimeError: Item video_0748 not found in the project.

annotation = ds.get_ann("video_0748.mp4", project.meta)
print(annotation.to_json())
# Output: {
#     "description": "",
#     "size": {
#         "height": 500,
#         "width": 700
#     },
#     "key": "e9ef52dbbbbb490aa10f00a50e1fade6",
#     "tags": [],
#     "objects": [],
#     "frames": [{
#         "index": 0,
#         "figures": []
#     }]
#     "framesCount": 1
# }

get_ann_path(item_name)[source]¶

Path to the given annotation json file.

Parameters

item_name : str: Item name.

Returns

Path to the given annotation json file.

Return type

str

Raises

RuntimeError if item not found in the project

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.get_ann_path("video_0748"))
# Output: RuntimeError: Item video_0748 not found in the project.

print(ds.get_ann_path("video_0748.mp4"))
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0/ann/video_0748.mp4.json'

get_image_info(item_name)[source]¶

Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

Return type: None

get_img_info_path(img_name)[source]¶

Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

Return type: str

get_img_path(item_name)[source]¶

Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

Return type: str

get_item_info(item_name)[source]¶

Information for Item with given name.

Parameters

item_name : str: Item name.

Returns

VideoInfo object.

Return type

VideoInfo

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.get_item_info("video_0748.mp4"))
# Output:
# VideoInfo(
#     id=198702499,
#     name='video_0748.mp4',
#     hash='ehYHLNFWmMNuF2fPUgnC/g/tkIIEjNIOhdbNLQXkE8Y=',
#     team_id=16087,
#     workspace_id=23821,
#     project_id=124974,
#     dataset_id=466639,
#     path_original='/h5un6l2bnaz1vj8a9qgms4-public/videos/w/7/i4/GZYoCs...9F3kyVJ7.mp4',
#     frames_to_timecodes=[0, 0.033367, 0.066733, 0.1001,...,10.777433, 10.8108, 10.844167],
#     frames_count=326,
#     frame_width=3840,
#     frame_height=2160,
#     created_at='2021-03-23T13:14:25.536Z',
#     updated_at='2021-03-23T13:16:43.300Z'
# )

get_item_info_path(item_name)[source]¶

Get path to the item info json file without checking if the file exists.

Parameters

item_name : str: Item name.

Returns

Path to the given item info json file.

Return type

str

Raises

RuntimeError if item not found in the project.

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.get_item_info_path("video_0748"))
# Output: RuntimeError: Item video_0748 not found in the project.

print(ds.get_item_info_path("video_0748.mp4"))
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0/video_info/video_0748.mp4.json'

get_item_path(item_name)[source]¶

Path to the given item.

Parameters

item_name : str: Item name.

Returns

Path to the given item.

Return type

str

Raises

RuntimeError if item not found in the project

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.get_item_path("video_0748"))
# Output: RuntimeError: Item video_0748 not found in the project.

print(ds.get_item_path("video_0748.mp4"))
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0/video/video_0748.mp4'

get_item_paths(item_name)[source]¶

Generates VideoItemPaths object with paths to item and annotation directories for item with given name.

Parameters

item_name : str: Item name.

Returns

VideoItemPaths object

Return type

VideoItemPaths

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

video_path, ann_path = dataset.get_item_paths("video_0748.mp4")
print("video_path:", video_path)
print("ann_path:", ann_path)
# Output:
# video_path: /home/admin/work/supervisely/projects/videos_example/ds0/video/video_0748.mp4
# ann_path: /home/admin/work/supervisely/projects/videos_example/ds0/ann/video_0748.mp4.json

get_items_names()[source]¶

List of video dataset item names.

Returns

List of item names.

Return type

list [ str ]

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.get_item_names())
# Output: ['video_0002.mp4', 'video_0005.mp4', 'video_0008.mp4', ...]

get_seg_path(item_name)[source]¶

Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

Return type: str

static get_url(project_id, dataset_id)[source]¶

Get URL to dataset items list in Supervisely.

Parameters

project_id : int: VideoProject ID in Supervisely.
dataset_id : int: VideoDataset ID in Supervisely.

Returns

URL to dataset items list.

Return type

str

Usage example

from supervisely import VideoDataset

project_id = 10093
dataset_id = 45330
ds_items_link = VideoDataset.get_url(project_id, dataset_id)

print(ds_items_link)
# Output: "/projects/10093/datasets/45330/entities"

item_exists(item_name)[source]¶

Checks if given item name belongs to the video dataset.

Parameters

item_name : str: Item name.

Returns

True if item exist, otherwise False.

Return type

bool

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

ds.item_exists("video_0748")     # False
ds.item_exists("video_0748.mp4") # True

items()¶

This method is used to iterate over dataset items, receiving item name, path to image and path to annotation json file. It is useful when you need to iterate over dataset items and get paths to images and annotations.

Returns

Generator object, that yields tuple of item name, path to image and path to annotation json file.

Return type

Generator[Tuple[str]]

Usage example

import supervisely as sly

input = "path/to/local/directory"
# Creating Supervisely project from local directory.
project = sly.Project(input, sly.OpenMode.READ)

for dataset in project.datasets:
    for item_name, image_path, ann_path in dataset.items():
        print(f"Item '{item_name}': image='{image_path}', ann='{ann_path}'")

set_ann(item_name, ann, key_id_map=None)[source]¶

Replaces given annotation for given item name to dataset annotations directory in json format.

Parameters

item_name : str: Item name.
ann : VideoAnnotation: VideoAnnotation object.
key_id_map : KeyIdMap, optional: KeyIdMap object.

Returns

None

Return type

NoneType

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

height, width = 500, 700
new_ann = sly.VideoAnnotation((height, width), frames_count=0)
ds.set_ann("video_0748.mp4", new_ann)

set_ann_dict(item_name, ann)¶

Replaces given annotation json for given item name to dataset annotations directory in json format.

Parameters

item_name : str: Item name.
ann : dict: Annotation as a dict in json format.

Returns

None

Return type

NoneType

Raises

RuntimeError if ann_path is not str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/lemons_annotated/ds1"
ds = sly.Dataset(dataset_path, sly.OpenMode.READ)

new_ann_json = {
    "description":"",
    "size":{
        "height":500,
        "width":700
    },
    "tags":[],
    "objects":[],
    "customBigData":{}
}

ds.set_ann_dict("IMG_8888.jpeg", new_ann_json)

set_ann_file(item_name, ann_path)¶

Replaces given annotation json file for given item name to dataset annotations directory in json format.

Parameters

item_name : str: Item Name.
ann_path : str: Path to the Annotation json file.

Returns

None

Return type

NoneType

Raises

RuntimeError if ann_path is not str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/lemons_annotated/ds1"
ds = sly.Dataset(dataset_path, sly.OpenMode.READ)

new_ann = "/home/admin/work/supervisely/projects/kiwi_annotated/ds1/ann/IMG_1812.jpeg.json"
ds.set_ann_file("IMG_1812.jpeg", new_ann)

property ann_dir¶

Path to the video dataset annotations directory.

Returns

Path to the video dataset directory with annotations.

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.ann_dir)
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0/ann'

ann_dir_name = 'ann'¶: str: Annotations directory name

property directory¶

Path to the video dataset directory.

Returns

Path to the video dataset directory.

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.directory)
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0'

property img_dir¶

Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

Return type: str

property img_info_dir¶: Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

property item_dir¶

Path to the video dataset items directory.

Returns

Path to the video dataset items directory.

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.item_dir)
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0/video'

item_dir_name = 'video'¶: str: Items data directory name

property item_info_dir¶

Path to the video dataset item with items info.

Returns

Path to the video dataset directory with items info.

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)

print(ds.item_info_dir)
# Output: '/home/admin/work/supervisely/projects/videos_example/ds0/video_info'

item_info_dir_name = 'video_info'¶: str: Items info directory name

property name¶

Video Dataset name.

Returns

Video Dataset Name.

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)
print(ds.name)
# Output: "ds0"

property project_dir¶

Path to the video project containing the video dataset.

Returns

Path to the video project.

Return type

str

Usage example

import supervisely as sly
dataset_path = "/home/admin/work/supervisely/projects/videos_example/ds0"
ds = sly.VideoDataset(dataset_path, sly.OpenMode.READ)
print(ds.project_dir)
# Output: "/home/admin/work/supervisely/projects/videos_example"

property seg_dir¶: Not available for VideoDataset class object. :raises: NotImplementedError in all cases.

seg_dir_name = None¶: str: Segmentation masks directory name