from custom.utils.visuals import merge_vggt
from vidar.utils.write import write_image
from vidar.utils.read import read_pickle


batch = read_pickle('batch.pkl')
predictions = read_pickle('predictions.pkl')

merge = merge_vggt(batch, predictions)
write_image('merge.png', merge)