File size: 2,250 Bytes
9279e0d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
# ------------------------------------------------------------------------
# Copyright (c) 2021 megvii-model. All Rights Reserved.
# ------------------------------------------------------------------------
# Modified from BasicSR (https://github.com/xinntao/BasicSR)
# Copyright 2018-2020 BasicSR Authors
# ------------------------------------------------------------------------
'''
for val set, extract the subset val-300
'''
import os
import time
from basicsr.utils.create_lmdb import create_lmdb_for_reds
def make_val_300(folder, dst):
if not os.path.exists(dst):
os.mkdir(dst)
templates = '*9.*'
cp_command = 'cp {} {}'.format(os.path.join(folder, templates), dst)
os.system(cp_command)
def flatten_folders(folder):
for vid in range(300):
vidfolder_path = '{:03}'.format(vid)
if not os.path.exists(os.path.join(folder, vidfolder_path)):
continue
print('working on .. {} .. {}'.format(folder, vid))
for fid in range(100):
src_filename = '{:08}'.format(fid)
suffixes = ['.jpg', '.png']
suffix = None
for suf in suffixes:
# print(os.path.join(folder, vidfolder_path, src_filename+suf))
if os.path.exists(os.path.join(folder, vidfolder_path, src_filename+suf)):
suffix = suf
break
assert suffix is not None
src_filepath = os.path.join(folder, vidfolder_path, src_filename+suffix)
dst_filepath = os.path.join(folder, '{}_{}{}'.format(vidfolder_path, src_filename, suffix))
os.system('mv {} {}'.format(src_filepath, dst_filepath))
time.sleep(0.001)
os.system('rm -r {}'.format(os.path.join(folder, vidfolder_path)))
if __name__ == '__main__':
flatten_folders('./datasets/REDS/train/train_blur_jpeg')
flatten_folders('./datasets/REDS/train/train_sharp')
# flatten_folders('./datasets/REDS/val/val_blur_jpeg')
# flatten_folders('./datasets/REDS/val/val_sharp')
# make_val_300('./datasets/REDS/val/val_blur_jpeg', './datasets/REDS/val/blur_300')
# make_val_300('./datasets/REDS/val/val_sharp', './datasets/REDS/val/sharp_300')
create_lmdb_for_reds()
|