Batch class for 2D images
@action
def load(self, src, fmt=None):
return super().load(src, fmt)
After Change
Load data
if fmt is None:
if isinstance(src, tuple):
self._data = tuple(src[i][self.indices] if len(src) > i else None for i in range(3))
else:
self._data = src[self.indices], None, None
else: