Video-Style-Transfer/config.py at dev · codermomo/Video-Style-Transfer · 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
66
67
68
69
70
71
72
73
IMG_CHANNELS = 3
DOWN_CHANNELS = [32, 48, 64]
NUM_RES = 5
UP_CHANNELS = [48, 32]
FINAL_CHANNELS = IMG_CHANNELS

VGG19_CFG = [
    64,
    64,
    "M",
    128,
    128,
    "M",
    256,
    256,
    256,
    256,
    "M",
    512,
    512,
    512,
    512,
    "M",
    512,
    512,
    512,
    512,
    "M",
]

VGG16_CFG = [
    64,
    64,
    "M",
    128,
    128,
    "M",
    256,
    256,
    256,
    "M",
    512,
    512,
    512,
    "M",
    512,
    512,
    512,
    "M",
]

LOSS_NET_CONFIG = {
    "VGG19": {
        "model_cfg": VGG19_CFG,
        "model_map": {
            "ReLU1_2": 3,
            "ReLU2_2": 8,
            "ReLU3_2": 13,
            "ReLU4_2": 22,
        },
        "content_layer": "ReLU4_2",
    },
    "VGG16": {
        "model_cfg": VGG16_CFG,
        "model_map": {
            "ReLU1_2": 3,
            "ReLU2_2": 8,
            "ReLU3_3": 15,
            "ReLU4_3": 22,
        },
        "content_layer": "ReLU2_2",
    },
}