d750d66fb998a44316000908f213ae73c53de97f,examples/shrec17/dataset.py,CacheNPY,__call__,#CacheNPY#,194

Before Change

            try: return np.load(npy_path.format(i))
            except OSError: exists[i] = False

        print("transform...", end="\r")
        try:
            img = self.transform(file_path)
        except:
            print(file_path)

After Change

            try: return np.load(npy_path.format(i))
            except OSError: exists[i] = False

        if self.pick_randomly:
            img = self.check_trans(file_path)
            np.save(npy_path.format(exists.index(False)), img)

            return img

        output = []
        for i in range(self.repeat):
            try:
                img = np.load(npy_path.format(i))
            except (OSError, FileNotFoundError):
                img = self.check_trans(file_path)
                np.save(npy_path.format(i), img)
            output.append(img)

        return output

    def __repr__(self):
        return self.__class__.__name__ + "(prefix={0}, transform={1})".format(self.prefix, self.transform)

Instances