dataset.py 文件源码

def rot(self):
        for i in range(len(self.Q)):
            Q       = self.Q[i]
            ans     = self.ans[i]
            refArea = self.area[i]
            refMask = self.mask[i]

            for rotate in range(3):
                self.Q.append(Q)
                self.ans.append(ans)

                refArea = np.rot90(refArea)
                refMask = np.rot90(refMask)

                self.area.append(refArea)
                self.mask.append(refMask)

        return self