diff --git a/ppgan/datasets/unpaired_dataset.py b/ppgan/datasets/unpaired_dataset.py
index 232f7bdbbecb3c2d2d8aebe76081156567e0816d..494099cdb46c920dc3181369f04dd6f03aa00329 100644
--- a/ppgan/datasets/unpaired_dataset.py
+++ b/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)
diff --git a/ppgan/modules/first_order.py b/ppgan/modules/first_order.py
index 77511719b64db874fd439a0e97cb91c974b76e73..3e84d8a4669078f8a6cb0bffdb93fbd79df254ac 100644
--- a/ppgan/modules/first_order.py
+++ b/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)
diff --git a/requirments.txt b/requirments.txt
index 51a8cf8ad94a9680d93b277d97fbdf4a3b6325a7..2816f6be1834c8b8dc10b9cd0a5de48ba1f9614c 100644
--- a/requirments.txt
+++ b/requirments.txt
@@ -2,3 +2,5 @@ tqdm
 PyYAML>=5.1
 scikit-image>=0.14.0
 scipy>=1.1.0
+opencv-python
+imageio-ffmpeg