update requirements.txt

ppgan/datasets/unpaired_dataset.py

@@ -64,8 +64,8 @@ class UnpairedDataset(BaseDataset):
             index_B = random.randint(0, self.B_size - 1)
         B_path = self.B_paths[index_B]
 
-        A_img = cv2.imread(A_path)
-        B_img = cv2.imread(B_path)
+        A_img = cv2.cvtColor(cv2.imread(A_path), cv2.COLOR_BGR2RGB)
+        B_img = cv2.cvtColor(cv2.imread(B_path), cv2.COLOR_BGR2RGB)
         # apply image transformation
         A = self.transform_A(A_img)
         B = self.transform_B(B_img)

ppgan/modules/first_order.py

@@ -118,7 +118,7 @@ class DownBlock2d(nn.Layer):
                               padding=padding,
                               groups=groups)
         self.norm = nn.BatchNorm2D(out_features)
-        self.pool = nn.AvgPool2d(kernel_size=(2, 2))
+        self.pool = nn.AvgPool2D(kernel_size=(2, 2))
 
     def forward(self, x):
         out = self.conv(x)

requirments.txt

@@ -2,3 +2,5 @@ tqdm
 PyYAML>=5.1
 scikit-image>=0.14.0
 scipy>=1.1.0
+opencv-python
+imageio-ffmpeg