使用svm进行蚊子识别

# -*- coding:utf-8 -*-
'''
@finish time:20170915
@author；fengjiexyb
this is a file that train and test LCD digit
the dataset is mine
the result is 95.05%
'''
import logging
import matplotlib.pyplot as plt
import numpy as np
import os
import pickle
import random
import struct
from PCV.tools import imtools
from PIL import Image
from svmutil import *
import cv2
Isize=75

#read  dataset
def load_data(path):
    files=os.listdir(path)
    features=[]
    label = []
    for index in files:
        #img = Image.open(path + index)
        img = cv2.imread(path+index)
        if img is not None:
            res = cv2.resize(img, (Isize, Isize), interpolation=cv2.INTER_AREA)
            im=cv2.cvtColor(res,cv2.COLOR_BGR2GRAY)
            #im = np.array(img.resize((50, 50)).convert('L'))
            im=np.reshape(im, Isize*Isize)
            features.append(im)