python - 将 cv2.umat 转换为 numpy 数组

Question

Processed_image() 函数返回一个 cv2.Umat 类型的值，该值将从 3 维(h, ch, w) 到 4 维(h, ch, w, 1) 所以 i 需要转换 到 numpy 数组或者如果可能的话帮助我直接重新整形 cv2.umat 要直接 reshape 并转换为 pytorch 张量的类型变量，可以是 分配给 reshaped_image_tensor。

img_w=640
img_h=640
img_ch=3
umat_img = cv2.UMat(img)
display_one(umat_img, "RESPONSE")    #function created by me to display image
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
with torch.no_grad():
    processed_img = preprocess_image(umat_img, model_image_size = (img_h, img_ch, img_w))
    #___________write YOUR CODE here________
    reshaped_images_tensor = torch.from_numpy(processed_img.reshape(img_h, img_ch, img_w, 1)).float().to(device)      #images_tensor.reshape(img_h, img_ch, img_w, 1)
    outputs = model(reshaped_images_tensor)
    _, predicted = torch.max(outputs, 1)
    c = predicted.squeeze()
    output_probability(predicted, processed_img, umat_img)
if ord('q')==cv2.waitKey(10):
    exit(0)

Answer 1

我没有完全理解你的问题，但你可以使用“get()”获取 opencv 的 umat 的 numpy 数据，例如 this

你可能应该permute你的张量，然后再将它输入你的模型。