Spaces:
Running
on
T4
Running
on
T4
File size: 1,037 Bytes
561c629 |
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 |
#!/bin/bash
# Set up the path (The following three paths will be kept and used in training)
full_patch_source=../APISR_dataset
degrade_hr_dataset_path=datasets/train_hr
train_hr_dataset_path=datasets/train_hr_enhanced
# tmp path (No need to change, we will remove them at the end of process)
tmp_dir_720p=APISR_720p_tmp
tmp_dir_720p_4xcrop=APISR_720p_crop_tmp
tmp_enhanced_dir=APISR_sharpen_tmp
# Resize images and prepare usm sharpening in Anime
python tools/720P_resize.py -i $full_patch_source -o $tmp_dir_720p
python tools/4x_crop.py -i $tmp_dir_720p -o $tmp_dir_720p_4xcrop
python scripts/anime_strong_usm.py -i $tmp_dir_720p_4xcrop -o $tmp_enhanced_dir --outlier_threshold 32
# Crop images to the target HR and degradate_HR dataset
python scripts/crop_images.py -i $tmp_dir_720p_4xcrop --crop_size 256 -o $degrade_hr_dataset_path
python scripts/crop_images.py -i $tmp_enhanced_dir --crop_size 256 -o $train_hr_dataset_path
# Clean unnecessary file
rm -rf $tmp_dir_720p
rm -rf $tmp_dir_720p_4xcrop
rm -rf $tmp_enhanced_dir |