From 39f81fce9075817c019915110d1964e7e774b714 Mon Sep 17 00:00:00 2001
From: ZGSLZL <45186889+ZGSLZL@users.noreply.github.com>
Date: Thu, 18 Jun 2020 10:47:07 +0800
Subject: [PATCH] Update casia.py

---
 dataset/casia.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dataset/casia.py b/dataset/casia.py
index 7ca636e..236d6e8 100644
--- a/dataset/casia.py
+++ b/dataset/casia.py
@@ -48,9 +48,9 @@ def reader():
                                         'profile', img_info['frames'][0])
                 label = img_info['labels'][0]
                 img = cv2.imread(img_path)
-                img, mask = self._get_mask(img, thr=2, crop=True)
-
                 img = img.astype(np.float32)
+                mask = np.zeros_like(img)
+                
                 img, mask, label = self.extra_aug(img, mask=mask, label=label)
 
                 flip = True if np.random.rand() < 0.5 else False