使用libsvm来训练液晶数字识别

最新推荐文章于 2023-04-26 00:05:01 发布

fengjiexyb

最新推荐文章于 2023-04-26 00:05:01 发布

阅读量1.2k

点赞数

分类专栏：图像识别 Python

本文链接：https://blog.csdn.net/fengjiexyb/article/details/78031109

版权

本文介绍了一次使用libsvm库进行液晶数字识别的训练过程。作者使用了自建的数据集，包含训练集130张（每个数字10张）和测试集101张。

摘要由CSDN通过智能技术生成

# -*- 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 *

#read  dataset
def load_data(path,i):
    files=os.listdir(path)
    features=[]
    label = []
    for index in files:
        image=Image.open(path+index)
        im=np.array(image.resize((28,28)).convert('L'))
        arr=255*np.ones((28,28),np.int)
        im=np.reshape(im, 784)
        features.append(im)
        if i=="train":
            label.append(int(index[4]))
        if i == "test":
            label.append(int(index[3]))
    return np.array(features),label,files
#add log
logging.basic