From 7e0305ed91fbb3a86ae5c598359190958fb5b056 Mon Sep 17 00:00:00 2001 From: Nils Koch Date: Thu, 2 Jun 2022 15:20:17 +0200 Subject: [PATCH] frontend/__init__.py: allow for data transfer minimization --- frontend/__init__.py | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/frontend/__init__.py b/frontend/__init__.py index 3940816..ef58e2a 100644 --- a/frontend/__init__.py +++ b/frontend/__init__.py @@ -44,7 +44,7 @@ def extract_data(data): # get result and rotate 90 deg pred_disp = cv2.transpose(np.asarray(data['disp'], dtype='uint8')) - if input not in data: + if 'input' not in data: return pred_disp, duration in_img = np.asarray(data['input'], dtype='uint8').transpose((2, 0, 1)) @@ -72,6 +72,12 @@ def put_image(img_path): return data +def change_minimal_data(enabled): + r = requests.post(f'{API_URL}/params/minimal_data/{not enabled}') + cv2.destroyWindow('Input Image') + cv2.destroyWindow('Reference Image') + + if __name__ == '__main__': while True: for img in os.scandir(img_dir): @@ -83,14 +89,18 @@ if __name__ == '__main__': downsize_input_img() data = put_image('buffer.png') - pred_disp, in_img, ref_pat, duration = extract_data(data) + if 'input' in data: + pred_disp, in_img, ref_pat, duration = extract_data(data) + else: + pred_disp, duration = extract_data(data) print(f'inference took {duration:1.4f}s') print(f'pipeline and transfer took another {(datetime.now() - start).total_seconds() - float(duration):1.4f}s') print(f"Pred. Disparity: \n\t{pred_disp.min():.{2}f}/{pred_disp.max():.{2}f}\n") - cv2.imshow('Input Image', in_img) - # cv2.imshow('Reference Image', ref_pat) + if 'input' in data: + cv2.imshow('Input Image', in_img) + cv2.imshow('Reference Image', ref_pat) cv2.imshow('Normalized Predicted Disparity', normalize_and_colormap(pred_disp)) cv2.imshow('Predicted Disparity', pred_disp) key = cv2.waitKey() @@ -99,3 +109,5 @@ if __name__ == '__main__': quit() elif key == 101: change_epoch() + elif key == 109: + change_minimal_data('input' not in data)