from data.DIT import dump_data as dump_data_image
from data.DIT import collect_data as collect_data_image
from data.DIT import load_data as load_data_image

if __name__ == '__main__':
    dump_data_image('data/superblue1_processed', 'data/superblue1', 900, 8, '100000', force_save=False)
    dump_data_image('data/superblue2_processed', 'data/superblue2', 893, 8, '100000', force_save=False)
    dump_data_image('data/superblue3_processed', 'data/superblue3', 800, 8, '100000', force_save=False)
    dump_data_image('data/superblue5_processed', 'data/superblue5', 853, 8, '100000', force_save=False)
    dump_data_image('data/superblue6_processed', 'data/superblue6', 857, 8, '100000', force_save=False)
    dump_data_image('data/superblue7_processed', 'data/superblue7', 800, 8, '100000', force_save=False)
    dump_data_image('data/superblue9_processed', 'data/superblue9', 800, 8, '100000', force_save=False)
    dump_data_image('data/superblue11_processed', 'data/superblue11', 900, 8, '100000', force_save=False)
    dump_data_image('data/superblue14_processed', 'data/superblue14', 800, 8, '100000', force_save=False)
    dump_data_image('data/superblue16_processed', 'data/superblue16', 700, 8, '100000', force_save=False)
    dump_data_image('data/superblue19_processed', 'data/superblue19', 900, 8, '100000', force_save=False)
    collect_data_image([
        'data/superblue1_processed',
        'data/superblue2_processed',
        'data/superblue3_processed',
        'data/superblue5_processed',
        'data/superblue6_processed',
        'data/superblue7_processed',
        'data/superblue9_processed',
        'data/superblue11_processed',
        'data/superblue14_processed',
    ], 'data/train_images', clear_files=True)
    collect_data_image([
        # 'data/superblue16_processed',
        'data/superblue19_processed',
    ], 'data/test_images', clear_files=False)
    _, train_images = load_data_image('data/train_images')
    _, test_images = load_data_image('data/test_images')
    print(len(train_images))
    print(len(test_images))
