Hi, Using PROBA-V dataset, the code crashes in lanczos file with thi

Lanczos code crashes about highres-net HOT 3 OPEN

servicenow commented on August 20, 2024 3

Lanczos code crashes

from highres-net.

Comments (3)

milder-z commented on August 20, 2024

I encountered the same problem.
And sadly, there seems to be no pre-trained model provided here.

from highres-net.

yunseok624 commented on August 20, 2024

Did someone solve the problem?

from highres-net.

jakubsadel commented on August 20, 2024

I thought that here it was enough to remove the transpose, because it looked like it disturbed the dimensionality, but nevertheless it did nothing :/.

def transform(self, theta, I, device="cpu"):
        '''
        Shifts images I by theta with Lanczos interpolation.
        Args:
            theta : tensor (B, 2), translation params
            I : tensor (B, C_in, H, W), input images
        Returns:
            out: tensor (B, C_in, W, H), shifted images
        '''

        self.theta = theta
        new_I = lanczos.lanczos_shift(img=I.transpose(0, 1), #delate transpose
                                      shift=self.theta.flip(-1),  # (dx, dy) from register_batch -> flip
                                      a=3, p=5)[:, None]
        return new_I

from highres-net.

Recommend Projects