iaraugen

Data augmentation/generation utilities for Iara.

Can be used standalone (instructions below) or as part of other programs.

Offline text augmentation

help: ./run_txt_aug.py -h

example usage:
./run_txt_aug.py corpus_1br_10pt_15sept.tok --aug backtranslate random_del --maxs 5 --lang en --translate_mode google --append --output out.txt

Offline text generation

help: ./run_txt_gen.py -h

example usage:
./run_txt_gen.py --input_file palavras.txt --context "radiologia médica" --num 2 --return_type "frases" --api_key "YOUR_OPENAI_API_KEY" --output query.txt

Offline audio augmentation

help: ./run_audio_aug.py -h 

example usage:
./run_audio_aug.py test.ogg --augmentations PitchShift GainTransition --output_format ogg

Corpus creation

help: ./run_create_corpus.py -h 

example usage:
./run_create_corpus.py dataset_es.txt --lang es --output corpus_es.tok