Skip to content

Commit

Permalink
Merge pull request #20 from ZGSLZL/master
Browse files Browse the repository at this point in the history
Update casia.py
  • Loading branch information
VIS-VAR authored Jun 18, 2020
2 parents d802d15 + 39f81fc commit bac93ab
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions dataset/casia.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit bac93ab

Please sign in to comment.