Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-09-16 18:15: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(
[
A.Flip(),
# A.ColorJitter(),
# A.ToGray(p=0.01),
# A.GaussianBlur(),
# A.MotionBlur(),
# A.ISONoise(),
# A.ImageCompression(),
# A.Normalize(
# mean=[0.485, 0.456, 0.406],
# std=[0.229, 0.224, 0.225],
# max_pixel_value=255,
# ), # [0, 255] -> coco (?) normalized
A.ToFloat(max_value=255),
A.ColorJitter(),
A.ToGray(p=0.01),
A.GaussianBlur(),
A.MotionBlur(),
A.ISONoise(),
A.ImageCompression(),
A.Normalize(
mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225],
max_pixel_value=255,
), # [0, 255] -> coco (?) normalized
ToTensorV2(), # HWC -> CHW
],
bbox_params=A.BboxParams(
@ -57,12 +56,11 @@ class Spheres(pl.LightningDataModule):
def val_dataloader(self):
transforms = A.Compose(
[
# A.Normalize(
# mean=[0.485, 0.456, 0.406],
# std=[0.229, 0.224, 0.225],
# max_pixel_value=255,
# ), # [0, 255] -> [0.0, 1.0] normalized
A.ToFloat(max_value=255),
A.Normalize(
mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225],
max_pixel_value=255,
), # [0, 255] -> [0.0, 1.0] normalized
ToTensorV2(), # HWC -> CHW
],
bbox_params=A.BboxParams(