-
Xavier Santos authored
Added new augmentation methods and the option of augmentation by addition instead of replacement (#653) * NOTEBOOK FOR DATA AUGMENTATION * SMALL SAMPLE DATASET * TRANSLACTIONS GLOBAL AND LOCAL * LOCAL AUGMENTATIONS FINISHED * ADJUSTED DATA FORMATS TO PCDET YAML COMPATIBILITY * BUGFIX * ADDED TITLES * ADAPTED OPENPCDET TO FIT NEW AUGMENTATIONS * SAVE IMAGES OF THE DATA AUGMENTATION EXAMPLES * LOCAL TRANSFORMATION RANDOM VALUES VARY WITH OBJECT AND NOT SCENE * FIXED YAML * FIXED BUGS IN PARAMETERS * FIX IN LOCAL BOXES * NEW BOX FILTER * FRUSTUM DROPOUT * FRUSTUM DROPOUT IMPLEMENTED ON OPENPCDET * REMOVE FIGURES FROM VERSION CONTROL * MORE EFFICIENT LOCAL METHOD * LOCAL TRANSFORMATION WITH NUMPY SPED UP * BUGFIX ON AUGMENTATION FUNCTION * GET PREDICTED LABELS * AUGMENTATION DONE BY ADDITION * RANDOM SELECTION OF REMAINIG FRAMES * LOGGER WITH TRANSFORMATION DETAILS * CHANGED COUNTER TO SAMPLE IDX SELECTION METHOD * BUGFIX ON KITTY_DATABASE * BUGFIX * BUGFIX * FIX ON GENERATOR * DEBUG LOGS TO FILE * FIXED LOGGER * SAVE POINT CLOUDS TO OBJ FILE * UPDATED NOTEBOOK * CLEANUP * .gitignore is now working * fixed .gitignore * REMOVED YAML FILES FROM .gitignore * parse results to a csv * yaml changed * fixed gitignore * FIXED BUG THAT DUPLICATED THE TEST SET * small fixes on yaml * fix gitignore * updated generate_sets.py * fixed csv_parser.py * Delete output directory * Delete output directory * saving figures with better resolution * implemented a way to chose to augment the raw pcs * updated csv_parser.py * fixed yamls to include the new selection system * notebook updated * mosaic added * new models * fixed gitignore * fixed gitignore * applied changes * Delete data_aumentation.ipynb * Delete pointpillar_fullaug.yaml * updated gitignore * bugfix * updated training and notebook * updated gitignore * cleanup code * updated yamls * reset * new augmentation methods * example conf for pointpillars * updated example config * bugfix on the config file * fixed get_points_in_box function * fixed local frustum dropout * fixed pointpillar_newaugs.yaml * bugfixed: import math for some augmentations * increased margin in point selection for local augmentation * bugfixed: velocity should not be translated for translation augs. Add warning to check the correctness before using * bugfixed: add back the aug of random_image_flip Co-authored-by:Shaoshuai Shi <shaoshuaics@gmail.com>
63ffbac9
