Skip to content

Commit a42de66

Browse files
committed
fix norm
1 parent e1678c6 commit a42de66

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

change_detection_pytorch/datasets/custom.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def get_default_transform(self):
154154

155155
default_transform = A.Compose([
156156
A.Resize(self.size, self.size),
157-
A.Normalize(mean=(0, 0, 0, 0, 0, 0), std=(1, 1, 1, 1, 1, 1)), # div(255)
157+
A.Normalize(),
158158
ToTensorV2()
159159
])
160160
return default_transform

0 commit comments

Comments
 (0)