You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
466 B
15 lines
466 B
import cv2
|
|
from cv2 import dnn_superres
|
|
# Create an SR object - only function that differs from c++ code
|
|
sr = dnn_superres.DnnSuperResImpl_create()
|
|
# Read image
|
|
image = cv2.imread('./images/person.jpg')
|
|
# Read the desired model
|
|
path = "EDSR_x4.pb"
|
|
sr.readModel(path)
|
|
# Set the desired model and scale to get correct pre- and post-processing
|
|
sr.setModel("edsr", 4)
|
|
# Upscale the image
|
|
result = sr.upsample(image)
|
|
# Save the image
|
|
cv2.imwrite("./upscaled.png", result) |