scripts/prepare_datasets.sh

#!/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