Closed faris-k closed 1 year ago
Albumentations probably won't work with lightly's collate functions, so to use albumentations over torchvision would probably require an entire rework of all the custom collate functions here. Instead, it would be better to re-implement whatever we need from albumentations as torchvision transforms. One that comes to mind is the OneOf
transform, which would be super useful for using DPWTransform
and DieNoise
(since both probably shouldn't be used at once, or they could but with a very low chance).
OneOf
to use generic torchvision transforms.
Minor enhancement, but since speed is a concern with self-supervised pretraining, use Albumentations instead of torchvision and lightly's transforms. Throughput is reportedly much higher: https://github.com/albumentations-team/albumentations#benchmarking-results