def get_result_img(self, input_left, input_right):
input_left_ = util.tensor2im(input_left[1, :, :, :])
input_right_ = util.tensor2im(input_right[1, :, :, :])
left_est_im = util.tensor2im(self.left_est[0][1, :, :, :])
right_est_im = util.tensor2im(self.right_est[0][1, :, :, :])
left_disp = util.tensor2im_(
self.disp_left_est[0][1, :, :, :].unsqueeze(0))
right_disp = util.tensor2im_(
self.disp_right_est[0][1, :, :, :].unsqueeze(0))
return OrderedDict([('input_left', input_left_),
('input_right', input_right_),
('left_est', left_est_im),
('right_est', right_est_im),
('left_disp', left_disp),
('right_disp', right_disp)])
def get_test_result(self, test_output):
self.test_output = test_output
test_output = util.tensor2im_(self.test_output[0, :, :, :])
return OrderedDict([('test_output', test_output)])