config_GAN.json

{
    "name": "RRDB_ESRGANx4",
    "n_gpu": 1,
    "model": "srgan",
    "distortion": "sr",
    "scale": 4,
    "use_tb_logger": true,

    "network_G": {
        "which_model_G": "RRDBNet",
        "in_nc": 3,
        "out_nc": 3,

        "nf": 64,
        "nb": 23,
        "args": {}
    },
    "network_D": {
        "which_model_G": "discriminator_vgg_128",
        "in_nc": 3,
        "nf": 64,
        "args": {}
    },
    "data_loader": {
        "type": "COWCGANFrcnnDataLoader",
        "args":{
            "data_dir_GT": "/home/jakaria/Super_Resolution/Datasets/COWC/DetectionPatches_256x256/Potsdam_ISPRS/HR/x4/",
            "data_dir_LQ": "/home/jakaria/Super_Resolution/Datasets/COWC/DetectionPatches_256x256/Potsdam_ISPRS/LR/x4/",
            "batch_size": 2,
            "shuffle": true,
            "validation_split": 0.0,
            "num_workers": 2
        }
    },
    "optimizer": {
        "type": "SGD",
        "args":{
        "lr_G": 0.0001,
        "weight_decay_G": 0,
        "beta1_G": 0.9,
        "beta2_G": 0.99,

        "lr_D": 0.0001,
        "weight_decay_D": 0,
        "beta1_D": 0.9,
        "beta2_D": 0.99
        }
    },
    "loss": "cross_entropy",
    "metrics": [
        "accuracy"
    ],
    "lr_scheduler": {
        "type": "MultiStepLR",
        "args": {
            "lr_steps": [50000, 100000, 200000, 300000],
            "lr_gamma": 0.5,
            "T_period": [250000, 250000, 250000, 250000],
            "restarts": [250000, 500000, 750000],
            "restart_weights": [1, 1, 1],
            "eta_min": 0.0000001
        }
    },
    "train": {
        "niter": 400000,
        "warmup_iter": -1,
        "pixel_criterion": "l1",
        "pixel_weight": 0.01,
        "feature_criterion": "l1",
        "feature_weight": 1,

        "gan_type": "ragan",
        "gan_weight": 0.001,
        "D_update_ratio": 1,
        "D_init_iters": 0,
        "manual_seed": 10,
        "val_freq": 1000,

        "save_dir": "saved/",
        "save_period": 1,
        "verbosity": 2,

        "monitor": "min val_loss",
        "early_stop": 10,

        "tensorboard": true
    },
    "path": {
        "models": "saved/pretrained_models_EESRGAN_FRCNN",
        "FRCNN_model": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/FRCNN_model_LR_LR_cowc/",
        "pretrain_model_G": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/pretrained_models_EESRGAN_FRCNN/170000_G.pth",
        "pretrain_model_D": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/pretrained_models_EESRGAN_FRCNN/170000_D.pth",
        "pretrain_model_FRCNN": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/pretrained_models_EESRGAN_FRCNN/170000_FRCNN.pth",
        "pretrain_model_FRCNN_LR_LR": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/FRCNN_model_LR_LR_cowc/0_FRCNN_LR_LR.pth",
        "training_state": "saved/training_state",
        "strict_load": true,
        "resume_state": "~",
        "val_images": "saved/val_images",
        "output_images": "saved/val_images_cars_new",
        "log": "saved/logs",
        "data_dir_Valid": "/home/jakaria/Super_Resolution/Datasets/COWC/DetectionPatches_256x256/Potsdam_ISPRS/LR/x4/valid_img/",
        "data_dir_F_SR": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/Final_SR_images_test/",
        "data_dir_SR": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/SR_images_test/",
        "data_dir_SR_combined": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/combined_SR_images_216000/",
        "data_dir_E_SR_1": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/enhanced_SR_images_1/",
        "data_dir_E_SR_2": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/enhanced_SR_images_2/",
        "data_dir_E_SR_3": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/enhanced_SR_images_3/",
        "data_dir_Bic": "/home/jakaria/Super_Resolution/Datasets/COWC/DetectionPatches_256x256/Potsdam_ISPRS/HR/x4/valid_img/",
        "data_dir_LR_train": "/home/jakaria/Super_Resolution/Datasets/COWC/DetectionPatches_256x256/Potsdam_ISPRS/LR/x4/",
        "data_dir_Bic_valid": "/home/jakaria/Super_Resolution/Datasets/COWC/DetectionPatches_256x256/Potsdam_ISPRS/Bic/x4/valid_img/",
        "Test_Result_LR_LR_COWC": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/Test_Result_LR_LR_COWC/",
        "Test_Result_SR": "/home/jakaria/Super_Resolution/Filter_Enhance_Detect/saved/Test_Result_SR/"
    },
    "logger": {
      "print_freq": 100,
      "save_checkpoint_freq": 1000
    }
}