Opencv 超分辨率错误：cv2.error: OpenCV(4.7.0) /io/opencv/modules/dnn/src.net_impl.cpp:1188

Question

我想将 Sentinel2 旋涡图像从 20m 升级到 10m，以将其与 r、g、b 和 nir 波段一起使用。 为此，我正在使用 python 代码尝试 opencv 超分辨率模块：

import cv2

img = cv2.imread("img.jp2")

sr = cv2.dnn_superres.DnnSuperResImpl_create()
sr.readModel("EDSR_x2.pb")
sr.setModel("edsr", 2)
result = sr.upsample(img)
cv2.imwrite("result.jp2", result)

使用 opencv 和 opencv-contrib 4.7.0.68，这个EDSR_x2.pb文件和这个图像

不幸的是，这一行产生了一个错误： result = sr.upsample(img)

错误：

cv2.error: OpenCV(4.7.0) /io/opencv/modules/dnn/src/net_impl.cpp:1188: error: (-2:Unspecified error) in function 'void cv::dnn::dnn4_v20221220::Net::Impl::getLayerShapesRecursively(int, cv::dnn::dnn4_v20221220::Net::Impl::LayersShapesMap&)'
>  (expected: 'total(os[i]) > 0'), where
>     'total(os[i])' is -874068992
> must be greater than
>     '0' is 0

我尝试将 opencv 和 opencv-contrib 降级到 4.6.0.66 和 4.5.5.64，结果相同。

有谁知道为什么它不起作用或有同样的问题？

Answer 1

您可以尝试从一个最小的工作示例开始：

import cv2


img = cv2.imread("img.png")
sr = cv2.dnn_superres.DnnSuperResImpl_create()
sr.readModel("models/EDSR_x2.pb")
sr.setModel("edsr", 2)
result = sr.upsample(img)
cv2.imwrite("out.png", result)

EDSR 文件： https://github.com/Saafke/EDSR_Tensorflow/tree/master/models

OpenCV：opencv-contrib-python-4.7.0.68

输入：

Output：