HobbitLong / SupContrast

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)
BSD 2-Clause "Simplified" License
3.12k stars 537 forks source link

What mean, std and other datapreprocessing transforms did you use for ImageNet? #144

Open LightingMc opened 10 months ago

LightingMc commented 10 months ago

It would really help me to know this information.

HobbitLong commented 10 months ago

Hi, @LightingMc ,

It has been a long time, but IIRC, the augmentation used is here.