a

Author: armarpc

mmseg/datasets/pipelines/hubmap_transforms.py

@@ -33,7 +33,7 @@ def aug_mega_hardcore(p=.95):
         ], p=.6),
         A.HueSaturationValue(p=.5),
         A.ChannelShuffle(p=.2),
-        A.ToTensor()
+        ToTensorV2()
     ], p=p)