如何将 Keras 与网络摄像头一起使用?
2020-07-27
1150
我有一个预训练模型。该模型使用 20000 个“灰色”样本进行训练。它正在使用“灰色”测试样本。但我想用网络摄像头测试这个模型。这是我的代码:
#Load the saved model
model = keras.models.load_model('C:\keras\handrecognition_model.h5')
video = cv2.VideoCapture(0)
while True:
_, frame = video.read()
im = Image.fromarray(frame, 'RGB')
im = im.resize((128, 128))
img_array = np.array(im)
img_array = np.expand_dims(img_array, axis=0)
prediction = int(model.predict(img_array)[0][0])
# if prediction is 0, which means I am missing on the image, then show the frame in gray color.
if prediction == 0:
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
cv2.imshow("Capturing", frame)
key = cv2.waitKey(1)
if key == ord('q'):
break
video.release()
cv2.destroyAllWindows()
出现错误:ValueError:检查输入时出错:预期 conv2d_1_input 具有形状(120, 320, 1),但得到的数组形状为(128, 128, 3)。
模型训练:
# Construction of model
model = Sequential()
model.add(Conv2D(32, (5, 5), activation='relu', input_shape=(120, 320, 1)))
model.add(MaxPooling2D((2, 2)))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D((2, 2)))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D((2, 2)))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(10, activation='softmax'))
编辑:我更新代码如下:
_, frame = video.read()
frame = cv2.resize(frame, (120, 360))
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
img_array = np.array(gray)
ValueError:检查输入时出错:预期 conv2d_1_input 有 4 个维度,但得到的数组形状为形状(1、360、120)
编辑 3:我猜,它正在工作。现在我将发送帧进行预测,我将找到手势。如果我可以做到,我会分享。谢谢。
_, frame = video.read()
frameCopy=frame.copy()
frameCopy = cv2.resize(frameCopy, (120, 320))
gray = cv2.cvtColor(frameCopy, cv2.COLOR_BGR2GRAY)
img_array = np.array(gray)
img_array = img_array.reshape(120, 320, 1)
img_array = np.expand_dims(img_array, axis=0)
2个回答
编辑后回答您的问题: 您需要 4 个维度,而不是 3 个:(批量大小、通道、宽度、高度)。因此,请尝试以下操作:
img_array = np.array(gray)
img_array = img_array.reshape(1, 1, 360, 120)
Theodor Peifer
2020-07-27
im = Image.fromarray(frame, 'RGB')
im = im.resize((128, 128))
im = im.convert('RGB')
这会将您的灰度图像转换为 RGB。它只会将值从您当前拥有的一维复制到三维。如果它抛出错误, 请尝试此方法
im = im.convert('L')
im = im.convert('RGB')
'L'
会先将其转换为黑白,以防您的输入有时不仅仅是 1D
theastronomist
2020-07-27