def __init__(
self,
doc,
dataset,
selected_fields=None,
excluded_fields=None,
filtered_fields=None,
):
if not isinstance(doc, foo.DatasetSampleDocument):
raise TypeError(
"Backing doc must be an instance of %s; found %s"
% (foo.DatasetSampleDocument, type(doc))
)
if not doc.id:
raise ValueError("`doc` is not saved to the database.")
if selected_fields is not None and excluded_fields is not None:
selected_fields = selected_fields.difference(excluded_fields)
excluded_fields = None
self._doc = doc
self._selected_fields = selected_fields
self._excluded_fields = excluded_fields
self._filtered_fields = filtered_fields
if self.media_type == fomm.VIDEO:
self._frames = fofr.Frames()
super().__init__(dataset=dataset)
def __next__(self):
"""Returns the next sample in the dataset.
Returns:
a :class:`fiftyone.core.sample.Sample`
Raises:
StopIteration: if there are no more samples to import
"""
d = next(self._iter_samples)
# Convert filepath to absolute path
d["filepath"] = os.path.join(self.dataset_dir, d["filepath"])
if self._is_video_dataset:
labels_relpath = d.pop("frames")
labels_path = os.path.join(self.dataset_dir, labels_relpath)
sample = fos.Sample.from_dict(d)
sample._frames = fof.Frames() # @todo clean up this hack
self._import_frame_labels(sample, labels_path)
else:
sample = fos.Sample.from_dict(d)
return sample
def __init__(self, filepath, tags=None, metadata=None, **kwargs):
self._doc = foo.NoDatasetSampleDocument(
filepath=filepath, tags=tags, metadata=metadata, **kwargs
)
if self.media_type == fomm.VIDEO:
self._frames = fofr.Frames()
super().__init__()
def from_doc(cls, doc, dataset=None):
"""Creates an instance of the :class:`Sample` class backed by the given
document.
Args:
doc: a :class:`fiftyone.core.odm.SampleDocument`
dataset (None): the :class:`fiftyone.core.dataset.Dataset` that
the sample belongs to
Returns:
a :class:`Sample`
"""
if isinstance(doc, foo.NoDatasetSampleDocument):
sample = cls.__new__(cls)
sample._dataset = None
sample._doc = doc
return sample
if not doc.id:
raise ValueError("`doc` is not saved to the database.")
try:
# Get instance if exists
sample = cls._instances[doc.collection_name][str(doc.id)]
except KeyError:
sample = cls.__new__(cls)
sample._doc = None # set to prevent RecursionError
if dataset is None:
raise ValueError(
"`dataset` arg must be provided for samples in datasets"
)
sample._set_backing_doc(doc, dataset=dataset)
if sample.media_type == fomm.VIDEO:
sample._frames = fofr.Frames()
return sample