settings_Rain200H_real.py

import os
import logging

patch_size = 128
pic_is_pair = True  # input picture is pair or single
ssim_loss = True
aug_data = False

lr = 1e-4
data_dir = '/media/ubuntu/Seagate/RainDataSet/JORDER/RainTrainH/'
if pic_is_pair is False:
    data_dir = '/data1/wangcong/dataset/real-world-images'
log_dir = './logdir_Rain200H_Real'
model_dir = ' ./trained_model/Rain200H+Real/'
show_dir_feature = '../showdir_feature'

log_level = 'info'
model_path = os.path.join(model_dir, 'net_latest')
save_steps = 400

num_workers = 8

num_GPU = 2
device_id = '0,1'

epoch = 10
batch_size = 2

if pic_is_pair:
    root_dir = os.path.join(data_dir, 'train')
    mat_files = os.listdir(root_dir)
    num_datasets = len(mat_files)
    l1 = int(3/5 * epoch * num_datasets / batch_size) # 90000
    l2 = int(4/5 * epoch * num_datasets / batch_size) # 120000
    one_epoch = int(num_datasets/batch_size)
    total_step = int((epoch * num_datasets)/batch_size)

logger = logging.getLogger('train')
logger.setLevel(logging.INFO)

ch = logging.StreamHandler()
ch.setLevel(logging.INFO)

formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
ch.setFormatter(formatter)
logger.addHandler(ch)