Learned-Lightweight-Smartphone-ISP-with-Unpaired-Data/config.py at main · chung-ting-huang/Learned-Lightweight-Smartphone-ISP-with-Unpaired-Data · GitHub

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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from easydict import EasyDict

config = EasyDict()

config.path_input = "../raw_images/train/mediatek_raw"
config.path_output = "../raw_images/train/fujifilm"
config.vgg_path = "../../vgg_pretrained/imagenet-vgg-verydeep-19.mat"
config.pretrained_path = "checkpoint/pretrained_20000.pth"
config.color_loss_type = "mse"
config.content_loss_type = 2
config.texture_loss_type = "relativistic"
config.shuffled = True
config.lambda_gp = 10
config.w_tv = 1.0
config.w_texture_factor = 1.0
config.w_color_factor = 1.0
config.w_content = 1.0
config.w_color = 1.0
config.w_texture = 1.0
config.w_dists = 1.0
config.w_lpips_plus = 1.0
config.chunks_count = 5000
config.batch_size = 32
config.num_batches = 100
config.chunks_size = 3200
config.gen_lr = 5e-4
config.gen_betas = [0.5, 0.9]
config.start_point = 1
config.gen_per_disc_ratio_color = 10
config.gen_per_disc_ratio_texture = 10
config.disc_grad_clip = 1
config.grayscale_channels = 1
config.rgb_channels = 3

config.discriminators = {

    "0": {
        "name": "Discriminator_LPIPS_64",
        "info": "LPIPS",
        "layer_id": 0,
        "learning_rate": 1e-5,
        "betas": (0.5, 0.9),
        "w_loss" : 1.0 * config.w_texture_factor,
        "channels" : 1
    },
    "3": {
        "name": "Discriminator_LPIPS_256",
        "info": "LPIPS",
        "layer_id": 3,
        "learning_rate": 1e-5,
        "betas": (0.5, 0.9),
        "w_loss" : 1.0 * config.w_texture_factor,
        "channels" : 1
    },
"5": {
        "name": "ViTDiscriminator",
        "info": "ViT",
        "layer_id": -1,
        "learning_rate": 1e-5,
        "betas": (0.5, 0.9),
        "w_loss": 1 * config.w_color_factor,
        "channels": 3
    }

}