update srgan

models/SRGAN/srgan-dl-prj.ipynb

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:4eb72ac34eeb3831604a783a87d65fe237d7ef6279dce4d99224ffa92f1211b3
-size 12001858

models/SRGAN/srgan.py

@@ -42,7 +42,7 @@ class GeneratorResnet(nn.Module):
             ]
         self.upsampling = nn.Sequential(*upsampling)
         
-        self.conv3 = nn.Sequential(nn.Conv2d(64, out_channels, kernel_size=9, stride=1, padding=4), nn.Tanh())
+        self.conv3 = nn.Conv2d(64, out_channels, kernel_size=9, stride=1, padding=4)
         
     def forward(self, x):
         out1 = self.conv1(x)
@@ -51,7 +51,7 @@ class GeneratorResnet(nn.Module):
         out = torch.add(out1, out2)
         out = self.upsampling(out)
         out = self.conv3(out)
-        return out.clamp(0, 1)
+        return out
     
     def inference(self, x):
         """
@@ -62,7 +62,7 @@ class GeneratorResnet(nn.Module):
             x = ToTensor()(x).unsqueeze(0)
             x = self.forward(x)
             x = Image.fromarray((x.squeeze(0).permute(1, 2, 0).detach().numpy() * 255).astype('uint8'))
-        return TF.adjust_brightness(x, 1.1)
+        return x
     
     def test(self, x):
         """

models/SRGAN/srgan_checkpoint.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9b1423b711b26b2250612e02fddf95a4a7214e883c601eb153bfa82caceb5511
-size 6336189
+oid sha256:252142cce821babfd7718b8a9ecb6c795e642d90fe4ccc69072c9317dc4a8982
+size 6336551