1.开发算法自动化创造数据集【训练】【验证】【测试】,人工过滤一波数据集
2.分类状态有三种,睁眼,闭眼,非眼【一开始用了两种,人眼和非人眼,对闭眼不能很好识别】
3.搭建cnn网络【采用类似VGG16的网络,但是网络一定要小,这样才能做到实时,方便后期ncnn移动端开发】
4.开发金字塔模型
5.对窗口用cnn predict
6.输入绘图,做可视化调试
1.自动化采集数据集算法【需要有摄像头,可用android手机 或者 usb 网络摄像头 或者视频,调节opencv参数打开】
# -*- coding: utf-8 -*-
"""
Created on Thu Sep 13 09:31:01 2018
@author: Lenovo
"""
import cv2
import numpy as np
import pylab as pl
from PIL import Image
import matplotlib.pyplot as plt
import time
pnum= list()
if __name__ == '__main__':
cap = cv2.VideoCapture(0)
iss=True
i=2000
while iss:
iss,img = cap.read()
i+=1
if iss:
# edgeimg = getGabor(img,filters)[0]
edgeimg &