Merge pull request #137 from ya0guang/patch-1

Seems like the `tmpLbl` variable is misused as

backgroundremover/u2net/data_loader.py

@@ -139,7 +139,7 @@ def __call__(self, sample):
         # 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))
-        tmpLbl = label.transpose((2, 0, 1))
+        tmpLbl = tmpLbl.transpose((2, 0, 1))
 
         return {
             "imidx": torch.from_numpy(imidx),