基于SuperResolution模型實(shí)現(xiàn)圖像超像素放大

omz_downloader --name single-image-super-resolution-1032

import cv2 as cvimport numpy as npimport openvino as ov
# Load the model on to the devicecore = ov.Core()
# Read the model.xml and weights filemodel = core.read_model(model="D:/projects/single-image-super-resolution-1032.xml")
compiled_model = core.compile_model(model=model, device_name="CPU")# Store the input and output nodesoriginal_image_key, bicubic_image_key = compiled_model.inputsoutput_layer = compiled_model.output(0)
bgr = cv.imread("D:/facedb/tiaoma/qr_blur_02.jpg")origin_bgr = cv.resize(src=bgr, dsize=(480, 270), interpolation=cv.INTER_CUBIC)bicubic_bgr = cv.resize(src=bgr, dsize=(1920, 1080), interpolation=cv.INTER_CUBIC)input_image = np.expand_dims(origin_bgr.transpose(2, 0, 1), axis=0)input_image_bicubic = np.expand_dims(bicubic_bgr.transpose(2, 0, 1), axis=0)
result = compiled_model({original_image_key.any_name: input_image,                        bicubic_image_key.any_name: input_image_bicubic})[output_layer]result = result.squeeze(0).transpose(1, 2, 0)result *= 255result[result < 0] = 0result[result > 255] = 255result = result.astype(np.uint8)
cv.imshow("super resolution", result)cv.imwrite("D:/result.jpg", result)cv.waitKey(0)cv.destroyAllWindows()