UMSN files

6 years ago · d5272892e3
parent 15718d86d2
commit d5272892e3
9 changed files with 7 additions and 1 deletions
--- a/datasets/init.pyc
+++ b/datasets/init.pyc
--- a/datasets/pix2pix.pyc
+++ b/datasets/pix2pix.pyc
--- a/datasets/pix2pix_class.pyc
+++ b/datasets/pix2pix_class.pyc
--- a/datasets/pix2pix_val.pyc
+++ b/datasets/pix2pix_val.pyc
--- a/myutils/utils.pyc
+++ b/myutils/utils.pyc
--- a/myutils/vgg16.py
+++ b/myutils/vgg16.py
@ -45,5 +45,11 @@ class Vgg16(torch.nn.Module):
        h = F.relu(self.conv4_2(h))
        h = F.relu(self.conv4_3(h))
        relu4_3 = h
+        h = F.max_pool2d(h, kernel_size=2, stride=2)
+
+        h = F.relu(self.conv5_1(h))
+        h = F.relu(self.conv5_2(h))
+        h = F.relu(self.conv5_3(h))
+        relu5_3 = h

-        return [relu1_2, relu2_2, relu3_3, relu4_3]
+        return [relu1_2, relu2_2, relu3_3, relu4_3, relu5_3]
--- a/myutils/vgg16.pyc
+++ b/myutils/vgg16.pyc
--- a/transforms/init.pyc
+++ b/transforms/init.pyc
--- a/transforms/pix2pix.pyc
+++ b/transforms/pix2pix.pyc