jqueguiner
commited on
Commit
·
dd6ddbd
1
Parent(s):
a8e0a60
init
Browse files- Face_Detection/shape_predictor_68_face_landmarks.dat +3 -0
- Face_Enhancement/checkpoints/Setting_9_epoch_100/latest_net_G.pth +3 -0
- Global/checkpoints/detection/FT_Epoch_latest.pt +3 -0
- Global/checkpoints/restoration/VAE_A_quality/latest_net_D.pth +3 -0
- Global/checkpoints/restoration/VAE_A_quality/latest_net_G.pth +3 -0
- Global/checkpoints/restoration/VAE_A_quality/latest_net_featD.pth +3 -0
- Global/checkpoints/restoration/VAE_A_quality/latest_optimizer_D.pth +3 -0
- Global/checkpoints/restoration/VAE_A_quality/latest_optimizer_G.pth +3 -0
- Global/checkpoints/restoration/VAE_A_quality/latest_optimizer_featD.pth +3 -0
- Global/checkpoints/restoration/VAE_B_quality/latest_net_D.pth +3 -0
- Global/checkpoints/restoration/VAE_B_quality/latest_net_G.pth +3 -0
- Global/checkpoints/restoration/VAE_B_quality/latest_optimizer_D.pth +3 -0
- Global/checkpoints/restoration/VAE_B_quality/latest_optimizer_G.pth +3 -0
- Global/checkpoints/restoration/VAE_B_scratch/latest_net_D.pth +3 -0
- Global/checkpoints/restoration/VAE_B_scratch/latest_net_G.pth +3 -0
- Global/checkpoints/restoration/VAE_B_scratch/latest_optimizer_D.pth +3 -0
- Global/checkpoints/restoration/VAE_B_scratch/latest_optimizer_G.pth +3 -0
- Global/checkpoints/restoration/mapping_quality/latest_net_D.pth +3 -0
- Global/checkpoints/restoration/mapping_quality/latest_net_mapping_net.pth +3 -0
- Global/checkpoints/restoration/mapping_quality/latest_optimizer_D.pth +3 -0
- Global/checkpoints/restoration/mapping_quality/latest_optimizer_mapping_net.pth +3 -0
- Global/checkpoints/restoration/mapping_scratch/iter.txt +2 -0
- Global/checkpoints/restoration/mapping_scratch/latest_net_D.pth +3 -0
- Global/checkpoints/restoration/mapping_scratch/latest_net_mapping_net.pth +3 -0
- Global/checkpoints/restoration/mapping_scratch/latest_optimizer_D.pth +3 -0
- Global/checkpoints/restoration/mapping_scratch/latest_optimizer_mapping_net.pth +3 -0
- Global/checkpoints/restoration/mapping_scratch/loss_log.txt +0 -0
- Global/checkpoints/restoration/mapping_scratch/model.txt +247 -0
Face_Detection/shape_predictor_68_face_landmarks.dat
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fbdc2cb80eb9aa7a758672cbfdda32ba6300efe9b6e6c7a299ff7e736b11b92f
|
3 |
+
size 99693937
|
Face_Enhancement/checkpoints/Setting_9_epoch_100/latest_net_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eaafe424c22225891f60e672f331e14521bef870e6ce490f988ce6ada4068569
|
3 |
+
size 368738923
|
Global/checkpoints/detection/FT_Epoch_latest.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b2d7ab04e9b3885c6b1991bb7a0b823129dd6e3ac078a9fd059ebd2a7ba59a95
|
3 |
+
size 451663663
|
Global/checkpoints/restoration/VAE_A_quality/latest_net_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9dc693d9430cdf62e6a2a0e140625ad43426e43b38302afab2720561e61629df
|
3 |
+
size 22121817
|
Global/checkpoints/restoration/VAE_A_quality/latest_net_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de39b6d0081746995afd2393a94a11a10ae052fd9484b4fb06b7fa6bd28dcf5d
|
3 |
+
size 3498887
|
Global/checkpoints/restoration/VAE_A_quality/latest_net_featD.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:24cfb20648f52d454f8ab3f6c31323e1ae46348334139141ccc16fbf817abf82
|
3 |
+
size 11310935
|
Global/checkpoints/restoration/VAE_A_quality/latest_optimizer_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23c154620cf57c9874b11baeb79406bc3f24afec7c1494046e83653c83d5dc84
|
3 |
+
size 44241647
|
Global/checkpoints/restoration/VAE_A_quality/latest_optimizer_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b9a84773dad2b877c69f514b5690353016da24803efb953a30bfdd3acb2b03b0
|
3 |
+
size 6990137
|
Global/checkpoints/restoration/VAE_A_quality/latest_optimizer_featD.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8e80ceba67a0a50149dc1b00636987a6e3f2e88555d612ae88a44b9a786a2b54
|
3 |
+
size 22620608
|
Global/checkpoints/restoration/VAE_B_quality/latest_net_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5badf02e2ee96f74636bc0c02b92b315b83c927d49bdbe7cb163613ed9c5dc5a
|
3 |
+
size 22121817
|
Global/checkpoints/restoration/VAE_B_quality/latest_net_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0e40d6ea7acf27f1420abd29197086c21d0c2a7b8fe98a9e09c60a339457a830
|
3 |
+
size 3498887
|
Global/checkpoints/restoration/VAE_B_quality/latest_optimizer_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d60fc7c0a6f5dfeaa2a98b4afb59a552ea41cc370c260ac640d12a7b64d71cc
|
3 |
+
size 44241647
|
Global/checkpoints/restoration/VAE_B_quality/latest_optimizer_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:603e46ce9aec73a0e385b90d3410cedfe5dcd3256ee4f7b12e1f6eee6d894fb3
|
3 |
+
size 6990137
|
Global/checkpoints/restoration/VAE_B_scratch/latest_net_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:63a72c03537bfe061f24d6497e995f544215fa768f204575646ad7ef195b903b
|
3 |
+
size 22121817
|
Global/checkpoints/restoration/VAE_B_scratch/latest_net_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e22bd7274229beaabf27427646d4623c7b5a302829dd9d466514ff639ba40f6a
|
3 |
+
size 3498887
|
Global/checkpoints/restoration/VAE_B_scratch/latest_optimizer_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:413edec10026fbd9b94f89b03d46c84e191bbea4d1c4e57acbc5d7d771ecbecb
|
3 |
+
size 44241647
|
Global/checkpoints/restoration/VAE_B_scratch/latest_optimizer_G.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e45d80c7ff0a6065be95ebdc583d962f222ba26de0af0ac0a610b637dcaae221
|
3 |
+
size 6990137
|
Global/checkpoints/restoration/mapping_quality/latest_net_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:619fb1391d32a930bfef1f09f36ed9eaa7868af32d827cb022cea50a25345d4c
|
3 |
+
size 22146393
|
Global/checkpoints/restoration/mapping_quality/latest_net_mapping_net.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bb193cc0c82a6f4ed1c746beec5366c8a1e4a1b8dad3c83148afd0f93814624b
|
3 |
+
size 144549781
|
Global/checkpoints/restoration/mapping_quality/latest_optimizer_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:776741c0d226674a5eb5b620c097f54da33d81cb41561a1e6b248cc711834c8b
|
3 |
+
size 44290799
|
Global/checkpoints/restoration/mapping_quality/latest_optimizer_mapping_net.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8089a1d0126784648649c2cf79ce517ff8a937526edaafaad6293090261fedb6
|
3 |
+
size 289093935
|
Global/checkpoints/restoration/mapping_scratch/iter.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
251
|
2 |
+
0
|
Global/checkpoints/restoration/mapping_scratch/latest_net_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6a3ebb451229f0ffcabeee7f5fadf749b758fc6f4c4f4847f4fa145826cc4a56
|
3 |
+
size 22281214
|
Global/checkpoints/restoration/mapping_scratch/latest_net_mapping_net.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a63f9e33802a262fa62b160785fa3c53166f3d1449046ad9afc742848e195578
|
3 |
+
size 205392667
|
Global/checkpoints/restoration/mapping_scratch/latest_optimizer_D.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b6aa331e4438c2845eb61b48a8483b8ce92a598b6a5ad39a26e22bbb136a8ac5
|
3 |
+
size 44290839
|
Global/checkpoints/restoration/mapping_scratch/latest_optimizer_mapping_net.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4ab84a0aacb687cb6fe35870bca2b27201a8a34ed1f753c1e0c8eba0865e278e
|
3 |
+
size 410776011
|
Global/checkpoints/restoration/mapping_scratch/loss_log.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
Global/checkpoints/restoration/mapping_scratch/model.txt
ADDED
@@ -0,0 +1,247 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
GlobalGenerator_DCDCv2(
|
2 |
+
(encoder): Sequential(
|
3 |
+
(0): ReflectionPad2d((3, 3, 3, 3))
|
4 |
+
(1): Conv2d(3, 64, kernel_size=(7, 7), stride=(1, 1))
|
5 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
6 |
+
(3): ReLU(inplace)
|
7 |
+
(4): Conv2d(64, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))
|
8 |
+
(5): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
9 |
+
(6): ReLU(inplace)
|
10 |
+
(7): Conv2d(64, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))
|
11 |
+
(8): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
12 |
+
(9): ReLU(inplace)
|
13 |
+
(10): ResnetBlock(
|
14 |
+
(conv_block): Sequential(
|
15 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
16 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
17 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
18 |
+
(3): ReLU(inplace)
|
19 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
20 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
21 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
22 |
+
)
|
23 |
+
)
|
24 |
+
(11): ResnetBlock(
|
25 |
+
(conv_block): Sequential(
|
26 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
27 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
28 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
29 |
+
(3): ReLU(inplace)
|
30 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
31 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
32 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
33 |
+
)
|
34 |
+
)
|
35 |
+
(12): ResnetBlock(
|
36 |
+
(conv_block): Sequential(
|
37 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
38 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
39 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
40 |
+
(3): ReLU(inplace)
|
41 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
42 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
43 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
44 |
+
)
|
45 |
+
)
|
46 |
+
(13): ResnetBlock(
|
47 |
+
(conv_block): Sequential(
|
48 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
49 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
50 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
51 |
+
(3): ReLU(inplace)
|
52 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
53 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
54 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
55 |
+
)
|
56 |
+
)
|
57 |
+
)
|
58 |
+
(decoder): Sequential(
|
59 |
+
(0): ResnetBlock(
|
60 |
+
(conv_block): Sequential(
|
61 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
62 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
63 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
64 |
+
(3): ReLU(inplace)
|
65 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
66 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
67 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
68 |
+
)
|
69 |
+
)
|
70 |
+
(1): ResnetBlock(
|
71 |
+
(conv_block): Sequential(
|
72 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
73 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
74 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
75 |
+
(3): ReLU(inplace)
|
76 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
77 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
78 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
79 |
+
)
|
80 |
+
)
|
81 |
+
(2): ResnetBlock(
|
82 |
+
(conv_block): Sequential(
|
83 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
84 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
85 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
86 |
+
(3): ReLU(inplace)
|
87 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
88 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
89 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
90 |
+
)
|
91 |
+
)
|
92 |
+
(3): ResnetBlock(
|
93 |
+
(conv_block): Sequential(
|
94 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
95 |
+
(1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
96 |
+
(2): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
97 |
+
(3): ReLU(inplace)
|
98 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
99 |
+
(5): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1))
|
100 |
+
(6): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
101 |
+
)
|
102 |
+
)
|
103 |
+
(4): ConvTranspose2d(64, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))
|
104 |
+
(5): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
105 |
+
(6): ReLU(inplace)
|
106 |
+
(7): ConvTranspose2d(64, 64, kernel_size=(4, 4), stride=(2, 2), padding=(1, 1))
|
107 |
+
(8): InstanceNorm2d(64, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
108 |
+
(9): ReLU(inplace)
|
109 |
+
(10): ReflectionPad2d((3, 3, 3, 3))
|
110 |
+
(11): Conv2d(64, 3, kernel_size=(7, 7), stride=(1, 1))
|
111 |
+
(12): Tanh()
|
112 |
+
)
|
113 |
+
)Mapping_Model_with_mask(
|
114 |
+
(before_NL): Sequential(
|
115 |
+
(0): Conv2d(64, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
116 |
+
(1): InstanceNorm2d(128, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
117 |
+
(2): ReLU(inplace)
|
118 |
+
(3): Conv2d(128, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
119 |
+
(4): InstanceNorm2d(256, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
120 |
+
(5): ReLU(inplace)
|
121 |
+
(6): Conv2d(256, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
122 |
+
(7): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
123 |
+
(8): ReLU(inplace)
|
124 |
+
(9): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
125 |
+
(10): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
126 |
+
(11): ReLU(inplace)
|
127 |
+
)
|
128 |
+
(NL): NonLocalBlock2D_with_mask_Res(
|
129 |
+
(g): Conv2d(512, 512, kernel_size=(1, 1), stride=(1, 1))
|
130 |
+
(W): Conv2d(512, 512, kernel_size=(1, 1), stride=(1, 1))
|
131 |
+
(theta): Conv2d(512, 512, kernel_size=(1, 1), stride=(1, 1))
|
132 |
+
(phi): Conv2d(512, 512, kernel_size=(1, 1), stride=(1, 1))
|
133 |
+
(res_block): Sequential(
|
134 |
+
(0): ResnetBlock(
|
135 |
+
(conv_block): Sequential(
|
136 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
137 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
138 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
139 |
+
(3): ReLU(inplace)
|
140 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
141 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
142 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
143 |
+
)
|
144 |
+
)
|
145 |
+
(1): ResnetBlock(
|
146 |
+
(conv_block): Sequential(
|
147 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
148 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
149 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
150 |
+
(3): ReLU(inplace)
|
151 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
152 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
153 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
154 |
+
)
|
155 |
+
)
|
156 |
+
(2): ResnetBlock(
|
157 |
+
(conv_block): Sequential(
|
158 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
159 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
160 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
161 |
+
(3): ReLU(inplace)
|
162 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
163 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
164 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
165 |
+
)
|
166 |
+
)
|
167 |
+
)
|
168 |
+
)
|
169 |
+
(after_NL): Sequential(
|
170 |
+
(0): ResnetBlock(
|
171 |
+
(conv_block): Sequential(
|
172 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
173 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
174 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
175 |
+
(3): ReLU(inplace)
|
176 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
177 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
178 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
179 |
+
)
|
180 |
+
)
|
181 |
+
(1): ResnetBlock(
|
182 |
+
(conv_block): Sequential(
|
183 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
184 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
185 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
186 |
+
(3): ReLU(inplace)
|
187 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
188 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
189 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
190 |
+
)
|
191 |
+
)
|
192 |
+
(2): ResnetBlock(
|
193 |
+
(conv_block): Sequential(
|
194 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
195 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
196 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
197 |
+
(3): ReLU(inplace)
|
198 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
199 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
200 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
201 |
+
)
|
202 |
+
)
|
203 |
+
(3): ResnetBlock(
|
204 |
+
(conv_block): Sequential(
|
205 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
206 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
207 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
208 |
+
(3): ReLU(inplace)
|
209 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
210 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
211 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
212 |
+
)
|
213 |
+
)
|
214 |
+
(4): ResnetBlock(
|
215 |
+
(conv_block): Sequential(
|
216 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
217 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
218 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
219 |
+
(3): ReLU(inplace)
|
220 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
221 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
222 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
223 |
+
)
|
224 |
+
)
|
225 |
+
(5): ResnetBlock(
|
226 |
+
(conv_block): Sequential(
|
227 |
+
(0): ReflectionPad2d((1, 1, 1, 1))
|
228 |
+
(1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
229 |
+
(2): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
230 |
+
(3): ReLU(inplace)
|
231 |
+
(4): ReflectionPad2d((1, 1, 1, 1))
|
232 |
+
(5): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1))
|
233 |
+
(6): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
234 |
+
)
|
235 |
+
)
|
236 |
+
(6): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
237 |
+
(7): InstanceNorm2d(512, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
238 |
+
(8): ReLU(inplace)
|
239 |
+
(9): Conv2d(512, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
240 |
+
(10): InstanceNorm2d(256, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
241 |
+
(11): ReLU(inplace)
|
242 |
+
(12): Conv2d(256, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
243 |
+
(13): InstanceNorm2d(128, eps=1e-05, momentum=0.1, affine=False, track_running_stats=False)
|
244 |
+
(14): ReLU(inplace)
|
245 |
+
(15): Conv2d(128, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
|
246 |
+
)
|
247 |
+
)
|