ailesanlesanleaile

image_outputs = model(image_inputs) # dict['est', 'gt', 'raw'] image_outputs = image_outputs['est'] # dist 1x1920x1024 image_outputs = image_outputs[0] # tensor 1x1920x1024 image_outputs = image_outputs.to("cpu").numpy().astype(np.uint8)[0] # ndarry 1920x1