Windows使用opencv训练模型过程记录（提供样本）

import os

from skimage.color import rgb2gray

import numpy as np

import skimage.io as io

import cv2

def convert_gray(f, **args):  # 图片处理与格式化的函数

    rgb = io.imread(f)  # 读取图片

    gray = rgb2gray(rgb)  # 将彩色图片转换为灰度图片

    dst = cv2.resize(gray, (50, 50))  # 调整大小，图像分辨率为40*40

    return dst

if __name__ == '__main__':

    '''

    批量转灰度图

    '''

    datapath = r'.'  # 图片所在的路径

    str = datapath + '/org/*.jpg'  # 识别.jpg的图像

    coll = io.ImageCollection(str, load_func=convert_gray)  # 批处理

    for i in range(len(coll)):

        io.imsave('./pos/' + np.str(i) + '.jpg', coll[i])  # 保存图片

import os

def create_pos_n_neg():

    for file_type in ['pos', 'neg1964']: #此处修改neg或pos即可

        if not os.path.exists(file_type):

            continue

        for img in os.listdir(file_type):

            if (file_type.startswith('neg')):

                line = 'D:/DEV_SDKs/Opencv/train3/'+file_type + '/' + img + '\n'

                with open('bg.txt', 'a') as f:

                    f.write(line)

            elif (file_type.startswith('pos')):

                line = 'D:/DEV_SDKs/Opencv/train5/'+file_type + '/' + img + ' 1 0 0 50 50\n'

                with open('info.txt', 'a') as f:

                    f.write(line)

if __name__ == '__main__':

    create_pos_n_neg()

    print('描述文件已生成！')

D:/DEV_SDKs/Opencv/train5/pos/0.jpg 1 0 0 50 50

D:/DEV_SDKs/Opencv/train5/pos/1.jpg 1 0 0 50 50

D:/DEV_SDKs/Opencv/train5/pos/10.jpg 1 0 0 50 50

...

D:/DEV_SDKs/Opencv/train5/neg/1.jpg

D:/DEV_SDKs/Opencv/train5/neg/10.jpg

D:/DEV_SDKs/Opencv/train5/neg/100.jpg

D:/DEV_SDKs/Opencv/train5/neg/11.jpg

...

opencv_createsamples.exe -vec info.vec -info info.txt -bg bg.txt -num 105 -w 50 -h 50

opencv_traincascade.exe -data data -vec info.vec -bg bg.txt -numPos 105 -numNeg 315 -numStages 20  -w 50 -h 50 -minHitRate 0.999 -maxFalseAlarmRate 0.5 -featureType LBP