AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

Fixs + Benchmark

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-02-21 15:29:34 -05:00
parent 49472adfab
commit f450fa8201
5 changed files with 60 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

14
higher/smart_aug/old/compare_TF.py Normal file
View file

@ -0,0 +1,14 @@
import RandAugment as rand
import PIL
import torchvision
import transformations as TF
tpil=torchvision.transforms.ToPILImage()
ttensor=torchvision.transforms.ToTensor()
img,label =data_train[0]
rimg=ttensor(PIL.ImageEnhance.Color(tpil(img)).enhance(1.5))#ttensor(PIL.ImageOps.solarize(tpil(img), 50))#ttensor(tpil(img).transform(tpil(img).size, PIL.Image.AFFINE, (1, -0.1, 0, 0, 1, 0)))#rand.augmentations.FlipUD(tpil(img),1))
timg=TF.color(img.unsqueeze(0),torch.Tensor([1.5])).squeeze(0)
print(torch.allclose(rimg,timg, atol=1e-3))
tpil(rimg).save('rimg.jpg')
tpil(timg).save('timg.jpg')