Currently using Windows 10 with Python 3.7 I tried running with this code in order to get the webcam to function for use in number recognition
def get_img_contour_thresh(img):
x, y, w, h = 0, 0, 300, 300
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
blur = cv2.GaussianBlur(gray, (5, 5), 0)
ret, thresh1 = cv2.threshold(blur, 175, 255, cv2.THRESH_BINARY_INV +
cv2.THRESH_OTSU)
thresh1 = thresh1[y:y + h, x:x + w]
contours, hierarchy = cv2.findContours(thresh1, cv2.RETR_TREE,
cv2.CHAIN_APPROX_SIMPLE)[-2:]
return img, contours, thresh1
def show_webcam(mirror=False):
# load json and create model
json_file = open('model.json', 'r')
loaded_model_json = json_file.read()
json_file.close()
model = model_from_json(loaded_model_json)
# load weights into new model
model.load_weights("model.h5")
print("Loaded model from disk")
model.compile(loss='categorical_crossentropy',
optimizer='adam',
metrics=['accuracy'])
cap = cv2.VideoCapture(0)
while True:
ret, img = cap.read()
img, contours, thresh = get_img_contour_thresh(img)
ans = ''
if len(contours) > 0:
contour = max(contours, key=cv2.contourArea)
if cv2.contourArea(contour) > 500:
x, y, w, h = cv2.boundingRect(contour)
newImage = thresh[y:y + h, x:x + w]
newImage = cv2.resize(newImage, (28, 28))
newImage = np.array(newImage)
newImage = newImage.astype('float32')
newImage /= 255
if K.image_data_format() == 'channels_first':
newImage = newImage.reshape(1, 28, 28)
else:
newImage = newImage.reshape(28, 28, 1)
newImage = np.expand_dims(newImage, axis=0)
ans = model.predict(newImage).argmax()
x, y, w, h = 0, 0, 300, 300
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 2)
cv2.putText(img, "CNN : " + str(ans), (10, 320),
cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2)
cv2.imshow("Frame", img)
cv2.imshow("Contours", thresh)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
def main():
show_webcam(mirror=True)
if __name__ == '__main__':
main()
I receive this error
File "cam.py", line 75, in <module>
main()
File "cam.py", line 71, in main
show_webcam(mirror=True)
File "cam.py", line 34, in show_webcam
img, contours, thresh = get_img_contour_thresh(img)
File "cam.py", line 10, in get_img_contour_thresh
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
cv2.error: OpenCV(3.4.4) C:\projects\opencv-
python\opencv\modules\imgproc\src\color.cpp:181: error: (-215:Assertion
failed) !_src.empty() in function 'cv::cvtColor'
Iv been looking for a solution for a while but havent had any luck in figuring this out, tried looking around but just cant figure it out, what do you guys think? Appreciate the help given