train.py

from utils import loadImages
from boostedcascade import BoostedCascade, HaarlikeFeature

GenerateFeatures = False
Database = 'x5large-2'
ModelFile = 'data/' + Database + '/model-100/' + Database

if __name__ == '__main__':
    # boostedCascade = BoostedCascade(0.03, 0.40, 0.99)
    # boostedCascade = BoostedCascade(0.04, 0.20, 0.985)
    # boostedCascade = BoostedCascade(0.07, 0.60, 0.97)

    if GenerateFeatures:
        boostedCascade = BoostedCascade(0.07, 0.60, 0.94)
        faceImages = loadImages('data/' + Database + '/train/faces')
        nonfaceImages = loadImages('data/' + Database + '/train/non-faces')

        boostedCascade.prepare(faceImages, nonfaceImages, shuffle=True, verbose=True)
        boostedCascade.savefeaturesdata('data/' + Database + '/train/features/' + Database)
    else:
        print('Loading model...')
        boostedCascade = BoostedCascade.loadModel(ModelFile)
        print(boostedCascade)
        print('Loading data...')
        boostedCascade.loadfeaturesdata('data/' + Database + '/train/features/' + Database)
        print('Training...')
        boostedCascade.train(is_continue=True, autosnap_filename=ModelFile, verbose=True)
        boostedCascade.saveModel(ModelFile)