# -*- 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)
使用svm进行蚊子识别
最新推荐文章于 2024-09-21 09:46:07 发布