-
Notifications
You must be signed in to change notification settings - Fork 0
/
stylefaster.py
41 lines (40 loc) · 1.27 KB
/
stylefaster.py
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
class FasterStyle:
def __init__(self, path):
self.path = path
self.beta = 1.0
self.style_conv1_1_check = False
self.style_conv1_1_weight = 0.2
self.style_conv1_2_check = True
self.style_conv1_2_weight = 0.2
self.style_pool1_check = False
self.style_pool1_weight = 0.2
self.style_conv2_1_check = False
self.style_conv2_1_weight = 0.2
self.style_conv2_2_check = True
self.style_conv2_2_weight = 0.2
self.style_pool2_check = False
self.style_pool2_weight = 0.2
self.style_conv3_1_check = False
self.style_conv3_1_weight = 0.2
self.style_conv3_2_check = False
self.style_conv3_2_weight = 0.2
self.style_conv3_3_check = True
self.style_conv3_3_weight = 0.2
self.style_pool3_check = False
self.style_pool3_weight = 0.2
self.style_conv4_1_check = False
self.style_conv4_1_weight = 0.2
self.style_conv4_2_check = False
self.style_conv4_2_weight = 0.2
self.style_conv4_3_check = True
self.style_conv4_3_weight = 0.2
self.style_pool4_check = False
self.style_pool4_weight = 0.2
self.style_conv5_1_check = False
self.style_conv5_1_weight = 0.2
self.style_conv5_2_check = False
self.style_conv5_2_weight = 0.2
self.style_conv5_3_check = True
self.style_conv5_3_weight = 0.2
self.style_pool5_check = False
self.style_pool5_weight = 0.2