Source code for sconce.data_generators.multi_class_image_data_generator

from sconce.data_generators import DataGenerator, ImageMixin

import pandas as pd


[docs]class MultiClassImageDataGenerator(DataGenerator, ImageMixin): """ An ImageDataGenerator class for use when each image may belong to more than one class. New in 0.10.0 """ def _get_class_df(self, targets=None): dataset = self.dataset rows = [] if targets is None: if hasattr(dataset, 'targets'): targets = dataset.targets else: raise RuntimeError("No targets were supplied, and the dataset doesn't " "have a 'targets' attribute") for target in targets: row = {_class: False for _class in dataset.classes} for idx in target: _class = dataset.classes[idx] row[_class] = True rows.append(row) return pd.DataFrame(rows)