16 lines
576 B
Python
16 lines
576 B
Python
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))
|