from ..src.dataloader import *

train_loader, val_loader = create_data_loaders(config['data_dir'])
batch = next(iter(train_loader))
google_img = batch['google_img'][0]
yandex_img = batch['yandex_img'][0]

# google_img.permute((1, 2, 0)) * 255
batch['homography_params'].mean(axis=0)

print(batch['homography_matrix'][0])
print(batch['homography_params'][0])
K = get_camera_matrix(config['image_size'][0], config['image_size'][1])
print(homography_params_to_matrix(batch['homography_params'][0], K))
print(matrix_to_homography_params(batch['homography_matrix'][0].numpy(), K))