【发布时间】:2021-09-27 08:58:53
【问题描述】:
import numpy as np
import cv2
import pickle
########### PARAMETERS ##############
width = 640
height = 480
threshold = 0.65 # MINIMUM PROBABILITY TO CLASSIFY
cameraNo = 1
#####################################
#### CREATE CAMERA OBJECT
cap = cv2.VideoCapture(cameraNo)
cap.set(3,width)
cap.set(4,height)
#### LOAD THE TRAINNED MODEL
pickle_in = open("model_trained.p","rb")
model = pickle.load(pickle_in)
print(cv2.__version__)
#### PREPORCESSING FUNCTION
def preProcessing(img):
img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
img = cv2.equalizeHist(img)
img = img/255
return img
while True:
success, imgOriginal = cap.read()
img = np.asarray(imgOriginal)
img = cv2.resize(img,(32,32)) <------ ERROR HERE
img = preProcessing(img)
cv2.imshow("Processsed Image",img)
img = img.reshape(1,32,32,1)
#### PREDICT
classIndex = int(model.predict_classes(img))
#print(classIndex)
predictions = model.predict(img)
#print(predictions)
probVal= np.amax(predictions)
print(classIndex,probVal)
if probVal> threshold:
cv2.putText(imgOriginal,str(classIndex) + " "+str(probVal),
(50,50),cv2.FONT_HERSHEY_COMPLEX,
1,(0,0,255),1)
cv2.imshow("Original Image",imgOriginal)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
错误:
文件“E:/DATA_SCIENCE/MY_WORK/test.py”,第 32 行,在
img = cv2.resize(img,(32,32)) cv2.error: OpenCV(4.5.3) :-1: error: (-5:Bad argument) in function 'resize' 重载解析失败:
- src 数据类型 = 17 不受支持
- 参数“src”的预期 Ptr<:umat>
【问题讨论】:
-
数据类型 17 表示
CV_8SC3 -
你想要这个结构。请参阅docs.opencv.org/4.1.1/da/d54/…,并且您的图像具有上述错误类型的数据。在 np.asarray() 中设置适当的数据类型。
标签: python image opencv image-resizing