Cpt.Captain/CREStereo-pytorch-nxt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

frontend/__init__.py: general improvements

Browse Source
This commit is contained in:
Nils Koch 2022-06-02 15:03:38 +02:00
parent 50581efa01
commit 922bd3ab80
1 changed files with 33 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
frontend/__init__.py
View File

@ -32,20 +32,23 @@ def normalize_and_colormap(img):
def change_epoch(): def change_epoch():
global r
epoch = input('Enter epoch number or "latest"\n') epoch = input('Enter epoch number or "latest"\n')
r = requests.post(f'{API_URL}/model/update/{epoch}') r = requests.post(f'{API_URL}/model/update/{epoch}')
print(r.text) print(r.text)
def extract_data(r): def extract_data(data):
# FIXME yuck, don't json the json # FIXME yuck, don't json the json
data = json.loads(json.loads(r.text)) duration = data['duration']
pred_disp = np.asarray(data['disp'], dtype='uint8')
# get result and rotate 90 deg
pred_disp = cv2.transpose(np.asarray(data['disp'], dtype='uint8'))
if input not in data:
return pred_disp, duration
in_img = np.asarray(data['input'], dtype='uint8').transpose((2, 0, 1)) in_img = np.asarray(data['input'], dtype='uint8').transpose((2, 0, 1))
ref_pat = np.asarray(data['reference'], dtype='uint8').transpose((2, 0, 1)) ref_pat = np.asarray(data['reference'], dtype='uint8').transpose((2, 0, 1))
duration = data['duration']
pred_disp = cv2.transpose(pred_disp)
return pred_disp, in_img, ref_pat, duration return pred_disp, in_img, ref_pat, duration
@ -65,10 +68,12 @@ def put_image(img_path):
r = requests.put(f'{API_URL}/ir', files=openBin) r = requests.put(f'{API_URL}/ir', files=openBin)
print('received response') print('received response')
r.raise_for_status() r.raise_for_status()
return r data = json.loads(json.loads(r.text))
return data
while True: if __name__ == '__main__':
while True:
for img in os.scandir(img_dir): for img in os.scandir(img_dir):
start = datetime.now() start = datetime.now()
if 'ir' not in img.path: if 'ir' not in img.path:
@ -77,8 +82,8 @@ while True:
# alternatively: use img.path for native size # alternatively: use img.path for native size
downsize_input_img() downsize_input_img()
r = put_image('buffer.png') data = put_image('buffer.png')
pred_disp, in_img, ref_pat, duration = extract_data(r) pred_disp, in_img, ref_pat, duration = extract_data(data)
print(f'inference took {duration:1.4f}s') 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'pipeline and transfer took another {(datetime.now() - start).total_seconds() - float(duration):1.4f}s')