pip install -r requirements.txt
python demo.py
demo.py
uses the sample data in data/
and generates GIF
outputs in output.
-
python main.py python main.py --data-root data/ --output-dir output/ --aug-prob 0.1
--data-root
: path to data directory--output-dir
: path to outputs directory--aug_prob
: probability with which each augmentation is applied, when the value is equal to1
, all augmentations are applied, and when the value is equal to0.1
an augmentation is applied with probability equal to0.1