DatasetType[source]

Enum = [IMAGE_CLASSIFICATION, IMAGE_SEGMENTATION, IMAGE_GENERATION, IMAGE_OBJECT_DETECTION]

Currently supported Machine Learning Types.

infer_dataset_type[source]

infer_dataset_type(category_file_path)

Try to infer the dataset type from the category file path. category_file_path: the path of the categories file return: the infered dataset type raises: ValueError if unsupported type infered