Skip to content

Conversation

@vdutell
Copy link

@vdutell vdutell commented May 5, 2022

No description provided.

anglemask = torch.from_numpy(anglemask).float().to(self.device)

# Bandpass filtering
banddft = lodft * anglemask * himask
Copy link

@ChairManMeow-SY ChairManMeow-SY May 30, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

should line 167 be:

banddft=(-1j)**order*lodft *anglemask * himask ?

As the code here

@ChairManMeow-SY
Copy link

I'm so confusing for line 91:
height, width = im_batch.shape[2], im_batch.shape[1]

As the input is in the shape [N, C, H, W], the squeezed im_batch should be in [N, H, W].....

Thus
height = im_batch.shape[1]
and
width = im_batch.shape[2] ?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants