fix model bug (PaddlePaddle#64)

AnnaTrainingG · Oct 30, 2020 · ac4b41e · ac4b41e
1 parent 1b84c4f
commit ac4b41e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/ppgan/datasets/single_dataset.py b/ppgan/datasets/single_dataset.py
@@ -47,7 +47,7 @@ def __getitem__(self, index):
             A_paths(str) - - the path of the image
         """
         A_path = self.A_paths[index]
-        A_img = cv2.imread(A_path)
+        A_img = cv2.cvtColor(cv2.imread(A_path), cv2.COLOR_BGR2RGB)
         A = self.transform(A_img)
 
         return {'A': A, 'A_paths': A_path}

diff --git a/ppgan/models/discriminators/nlayers.py b/ppgan/models/discriminators/nlayers.py
@@ -135,8 +135,7 @@ def __init__(self, input_nc, ndf=64, n_layers=3, norm_type='instance'):
                           1,
                           kernel_size=kw,
                           stride=1,
-                          padding=padw,
-                          bias_attr=False)
+                          padding=padw)
             ]  # output 1 channel prediction map
 
         self.model = nn.Sequential(*sequence)