neodroidvision.data.detection.voc.voc.VOCDataset¶
- class neodroidvision.data.detection.voc.voc.VOCDataset(data_root: Path, dataset_name: str, split: SplitEnum, img_transform: Optional[callable] = None, annotation_transform: Optional[callable] = None)[source]¶
Bases:
ObjectDetectionDataset
description
- __init__(data_root: Path, dataset_name: str, split: SplitEnum, img_transform: Optional[callable] = None, annotation_transform: Optional[callable] = None)[source]¶
Dataset for VOC data.
data_root: the root of the VOC2007 or VOC2012 dataset, the directory contains the following
Annotations, ImageSets, JPEGImages, SegmentationClass, SegmentationObject.
- Parameters
data_root –
dataset_name –
split –
img_transform –
annotation_transform –
keep_difficult –
Methods
__init__
(data_root, dataset_name, split[, ...])Dataset for VOC data.
get_annotation
(index)- param index
get_img_info
(index)- param index
Attributes
categories
data_dirs
description
- param self
return: :rtype:
splits