easy_vision.python.data_main

Binary to prepare data for easy vision model.

easy_vision.python.data_main.create_dataset(input_path, config_path, output_prefix)

convert input_path file into tf record file

Parameters:

• input_path – raw input file path
• config_path – convert config file path
• output_prefix – output prefix to save tfrecord

Raises:

AssertionError – if config_path does not exist or config_path.proc_num <= 0

easy_vision.python.data_main.create_dataset_with_param_config(input_path, param_config_str, output_prefix, data_prefix='')

convert input_path file into tf record file with param config

Parameters:

• input_path – raw input file path
• param_config_str – parameter config str
• output_prefix – output prefix to save tfrecord
• data_prefix – data path prefix for data and pretrained-models, e.g., osspath oss://pai-vision-data/, local path /home/user/data

Raises:

AssertionError – if config_path does not exist or config_path.proc_num <= 0