IBM_AI/vi-opton/opton_od_train/template/make_data_files.py

# -*- coding=utf-8 -*-

import os
import argparse

from vi_od_frcnn.data.rcnn_data_util import make_data_files

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--data_dir', help='rcnn dataset dir',
                        default='./data', type=str)
    parser.add_argument('--train_percent', help='percent for train images (train+val = all * train_percent)',
                        default=0.9, type=float)
    parser.add_argument('--val_percent',  help='percent for val images (val = (train+val) * val_percent)',
                        default=0.1, type=float)
    parser.add_argument('--split_str',
                        default=None, type=str)
    args = parser.parse_args()
    return args

if __name__ == '__main__':

    args = parse_args()
    assert os.path.exists(args.data_dir), args.data_dir + 'does not exist'

    make_data_files(args.data_dir, args.train_percent, args.val_percent, args.split_str)