Python之手写数字识别

import cv2 import os from img_str import verify_img_str # 创建窗口，设置为不能改变大小 cv2.namedWindow('camera', cv2.WINDOW_NORMAL) # 1. 修改窗口属性，允许改变大小 # 摄像头设置，这里使用的是网络摄像头的URL video = 'http://admin:admin@192.168.1.103:8081' capture = cv2.VideoCapture(video) # 2. 初始化摄像头 # 初始化保存图片的序号 p = 1 # 开始无限循环，直到用户中断 while True: success, img = capture.read() # 3. 从摄像头读取一帧图像 if not success: print("无法抓取帧") # 4. 如果读取失败，打印错误信息并退出循环 break # 将图像旋转90度以竖直显示 img = cv2.rotate(img, cv2.ROTATE_90_CLOCKWISE) # 5. 旋转图像 # 显示图像 cv2.imshow("camera", img) # 6. 在窗口中显示图像 # 按键处理 key = cv2.waitKey(10) # 7. 等待用户按键，10毫秒响应一次 if key == 27: # 8. 如果按下ESC键（ASCII码为27），退出循环 break elif key == ord(' '): # 9. 如果按下空格键（ASCII码为32），保存图片 # 保存图片到当前目录下的img文件夹中 img_folder = 'img' if not os.path.exists(img_folder): os.makedirs(img_folder) # 10. 如果不存在img文件夹，则创建 img_filename = f"{img_folder}/image_{p}.jpg" # 11. 生成图片文件名 cv2.imwrite(img_filename, img) # 12. 保存图片 print(f"Saved image {p}.jpg") result = verify_img_str(file_path=img_filename) print(result) p += 1 # 13. 更新图片序号 # 释放摄像头资源 capture.release() # 关闭所有OpenCV创建的窗口 cv2.destroyAllWindows()