Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-09-19 11:35:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: augmentations

Browse source 
Former-commit-id: d06568c40226cb9b2e4f80b4fe1dea1f094a5934 [formerly ecc833dfd23fc9065637e24de9b6ce94410f5c10]
Former-commit-id: 70d75d9c26a3010618a5b6e42ca197d1917ef640
This commit is contained in:
Laurent Fainsin 2022-09-07 10:43:12 +02:00
parent 56e24615e3
commit 3bbc132359
1 changed files with 16 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src/data/dataloader.py
View file

@ -19,18 +19,17 @@ class Spheres(pl.LightningDataModule):
transforms = A.Compose( transforms = A.Compose(
[ [
A.Flip(), A.Flip(),
# A.ColorJitter(), A.ColorJitter(),
# A.ToGray(p=0.01), A.ToGray(p=0.01),
# A.GaussianBlur(), A.GaussianBlur(),
# A.MotionBlur(), A.MotionBlur(),
# A.ISONoise(), A.ISONoise(),
# A.ImageCompression(), A.ImageCompression(),
# A.Normalize( A.Normalize(
# mean=[0.485, 0.456, 0.406], mean=[0.485, 0.456, 0.406],
# std=[0.229, 0.224, 0.225], std=[0.229, 0.224, 0.225],
# max_pixel_value=255, max_pixel_value=255,
# ), # [0, 255] -> coco (?) normalized ), # [0, 255] -> coco (?) normalized
A.ToFloat(max_value=255),
ToTensorV2(), # HWC -> CHW ToTensorV2(), # HWC -> CHW
], ],
bbox_params=A.BboxParams( bbox_params=A.BboxParams(
@ -57,12 +56,11 @@ class Spheres(pl.LightningDataModule):
def val_dataloader(self): def val_dataloader(self):
transforms = A.Compose( transforms = A.Compose(
[ [
# A.Normalize( A.Normalize(
# mean=[0.485, 0.456, 0.406], mean=[0.485, 0.456, 0.406],
# std=[0.229, 0.224, 0.225], std=[0.229, 0.224, 0.225],
# max_pixel_value=255, max_pixel_value=255,
# ), # [0, 255] -> [0.0, 1.0] normalized ), # [0, 255] -> [0.0, 1.0] normalized
A.ToFloat(max_value=255),
ToTensorV2(), # HWC -> CHW ToTensorV2(), # HWC -> CHW
], ],
bbox_params=A.BboxParams( bbox_params=A.BboxParams(