Source code for sconce.data_feeds.multi_class_image

from sconce.data_feeds import ImageFeed

import pandas as pd


[docs]class MultiClassImageFeed(ImageFeed): """ An ImageFeed class for use when each image may belong to more than one class. """ def _get_class_df(self, dataset=None): if dataset is None: dataset = self.dataset rows = [] for target in dataset.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)