64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
# ------------------------------------------------------------------------
|
|
# 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()
|
|
|
|
|