issue-6: added conditional image augmentation (triggered by --augment_images)

arampacha / CLIP-rsicd

Apache License 2.0

212 stars 31 forks source link

Closed sujitpal closed 3 years ago

sujitpal commented 3 years ago

Changes:

run_clip_flax.py
- added a new parameter data_args.augment_images
- added a Pytorch transform for image augmentation
- have the image augmentation transform be triggered by augment_images parameter.
run_clip_training.sh -- added a augment_images parameter (currently commented out).