ndsampler.coco_frames module

class ndsampler.coco_frames.CocoFrames(dset, hashid_mode='PATH', workdir=None, verbose=0, backend='auto')

Bases: Frames, HashIdentifiable

wrapper around coco-style dataset to allow for getitem syntax

CommandLine

xdoctest -m ndsampler.coco_frames CocoFrames

Example

>>> from ndsampler.coco_frames import *
>>> import ndsampler
>>> import kwcoco
>>> import ubelt as ub
>>> workdir = ub.Path.appdir('ndsampler').ensuredir()
>>> dset = kwcoco.CocoDataset.demo(workdir=workdir)
>>> dset._ensure_imgsize()
>>> self = CocoFrames(dset, workdir=workdir)
>>> assert self.load_image(1).shape == (512, 512, 3)
>>> assert self.load_image(1)[:-20, :-10].shape == (492, 502, 3)
>>> assert self.load_region(1, (slice(-20), slice(-10))).shape == (492, 502, 3)

Example

>>> from ndsampler import coco_sampler
>>> self = coco_sampler.CocoSampler.demo().frames
>>> assert self.load_image(1).shape == (600, 600, 3)
>>> assert self.load_image(1)[:-20, :-10].shape == (580, 590, 3)

property image_ids

load_region(image_id, region=None, channels=NoParam)