19a19478d2dc1cdff7321f156512f66dbd6c5dd6,dataset/batch.py,ImagesBatch,load,#ImagesBatch#Any#Any#,224

Before Change

     Batch class for 2D images 
    @action
    def load(self, src, fmt=None):
        return super().load(src, fmt)

After Change

    @action
    def load(self, src, fmt=None):
         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:
            raise ValueError("Unsupported format:", fmt)
        return self

    @action
    def dump(self, dst, fmt=None):

Instances