gidariss / FeatureLearningRotNet

MIT License
508 stars 120 forks source link

Problem in data transformations #12

Open rabiaali95 opened 4 years ago

rabiaali95 commented 4 years ago

In dataloader.py when preparing rotated images rotated_imgs = [

                self.transform(img0),
                self.transform(rotate_img(img0,  90)),
                self.transform(rotate_img(img0, 180)),
                self.transform(rotate_img(img0, 270))
            ]

the following error arises. ValueError: some of the strides of a given numpy array are negative. This is currently not supported, but will be added in future releases.

How to solve this error?

rxqy commented 4 years ago

Hi, a simple solution would be to add a .copy() to all of the output of rotate_img

From https://github.com/gidariss/FeatureLearningRotNet/blob/817342ba8521ef709467c8bb7e3ec157969c1145/dataloader.py#L216-L226

To return np.flipud(...),copy() return np.fliplr(...).copy() return np.transpose(...).copy()

For more details of this problem, you can check this thread from the pytorch form