AI/U-2-Net
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net synced 2025-08-18 13:05:55 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

new

Browse Source
This commit is contained in:
Nathan Qin 2020-05-27 21:32:59 -06:00
parent 2467db76f7
commit 10b9b6767f
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
data_loader.py
View File

@ -85,6 +85,10 @@ class RandomCrop(object):
def __call__(self,sample): def __call__(self,sample):
imidx, image, label = sample['imidx'], sample['image'], sample['label'] imidx, image, label = sample['imidx'], sample['image'], sample['label']
if random.random() >= 0.5:
image = image[::-1]
label = label[::-1]
h, w = image.shape[:2] h, w = image.shape[:2]
new_h, new_w = self.output_size new_h, new_w = self.output_size