Esempio n. 1
0
def main(img_dir, img_name, classes):
    config = Config()
    config.batch_size = 1
    config.num_classes = 2

    result = {}
    for index in classes:
        result[index] = predict(img_dir + img_name, index, config)
    print(result)
    return result