given
$$
| import torch, kornia, h5py, imageio | |
| import matplotlib.pyplot as plt | |
| import numpy as np | |
| # This is a failure mode, with different aspect ratios | |
| fname1 = '47698078_3766965066' | |
| fname2 = '18698491_4586522698' | |
| # This works | |
| # fname1 = '271147142_778c4e7999_o' |
| from __future__ import division | |
| from __future__ import print_function | |
| import argparse | |
| import math | |
| import time | |
| from tqdm import trange | |
| import torch | |
| from correlation_package.correlation import Correlation |
| import time | |
| import torch | |
| import torch.backends.cudnn as cudnn | |
| import torch.nn.functional as F | |
| from torch.autograd import Variable | |
| def benchmark_shape(N, C, IH, IW, H, W, nrand, nrep): | |
| """ |
given
$$