|
|
7 anos atrás | |
|---|---|---|
| .. | ||
| mixins | 7 anos atrás | |
| README.md | 7 anos atrás | |
| __init__.py | 7 anos atrás | |
| image.py | 7 anos atrás | |
| utils.py | 7 anos atrás | |
The provided Dataset class inherits from the AnnotationsReadMixin and the PartMixing.
Consequently, to create a Dataset instance an Annotation object is required (it is necessary to pass the names of the arguments!):
# replace NAB_Annotations with CUB_Annotations to load CUB200-2011 annotations
from nabirds import NAB_Annotations, Dataset
annot = NAB_Annotations("path/to/nab/folder")
train_data = Dataset(uuids=annot.train_uuids, annotations=annot)
test_data = Dataset(uuids=annot.test_uuids, annotations=annot)
In order to enable the PartMixing functionalities pass the appropriate arguments to the initializer:
data = Dataset(
uuids=uuids, annotations=annot,
uniform_parts=True,
crop_to_bb=True,
crop_uniform=True,
parts_in_bb=True,
rnd_select=True,
ratio=0.35,
seed=21486214
)
Information about all possible PartMixing functionalities can be found here.
All of the used part and bounding box functionalities can be manually used with the help of the utils module.
utils moduleTBD (see main page for more information for now)