AI/U-2-Net
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net synced 2025-07-31 01:52:03 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update data_loader.py

Browse Source
This commit is contained in:
Xuebin Qin 2021-12-28 11:05:13 +04:00 committed by GitHub
parent 1948f813aa
commit a179b4bfd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
data_loader.py
View File

@ -127,8 +127,7 @@ class ToTensor(object):
tmpLbl[:,:,0] = label[:,:,0] tmpLbl[:,:,0] = label[:,:,0]
# change the r,g,b to b,r,g from [0,255] to [0,1]
#transforms.Normalize(mean = (0.485, 0.456, 0.406), std = (0.229, 0.224, 0.225))
tmpImg = tmpImg.transpose((2, 0, 1)) tmpImg = tmpImg.transpose((2, 0, 1))
tmpLbl = label.transpose((2, 0, 1)) tmpLbl = label.transpose((2, 0, 1))
@ -215,8 +214,7 @@ class ToTensorLab(object):
tmpLbl[:,:,0] = label[:,:,0] tmpLbl[:,:,0] = label[:,:,0]
# change the r,g,b to b,r,g from [0,255] to [0,1]
#transforms.Normalize(mean = (0.485, 0.456, 0.406), std = (0.229, 0.224, 0.225))
tmpImg = tmpImg.transpose((2, 0, 1)) tmpImg = tmpImg.transpose((2, 0, 1))
tmpLbl = label.transpose((2, 0, 1)) tmpLbl = label.transpose((2, 0, 1))