python - 基于 CNN 模型预测值的车牌裁剪图像

Question

我开发了一个 DNN 模型来预测车牌的位置。然后我想获取预测的矩形并根据其在图像中的位置裁剪图像。

当前错误：

TypeError: only integer scalar arrays can be converted to a scalar index

当前代码：

import cv2 
import imutils
import numpy as np
import pytesseract
pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files (x86)\Tesseract-OCR\tesseract.exe'

img = cv2.resize(cv2.imread("licensed_car109.jpeg",cv2.IMREAD_COLOR) / 255, dsize=(WIDTH, HEIGHT))

y_hat = model.predict(img.reshape(1, WIDTH, HEIGHT, 3)).reshape(-1) * WIDTH

xt, yt = y_hat[0], y_hat[1]
xb, yb = y_hat[2], y_hat[3]

img = cv2.cvtColor(img.astype(np.float32), cv2.COLOR_BGR2RGB)
image = cv2.rectangle(img, (xt, yt), (xb, yb), (0, 0, 255), 1)
plt.imshow(image)
plt.show()

Cropped = img[x:(xt, xb), y:(yt, yb)]

标签： pythonarraysimagenumpyconv-neural-network