GitLab

refactor database_sampler.py to support extending to other multimodal datasets (in addition to kitti)

* add pyramid augmentation

* add pyramid augmentation

* format the modified file
Co-authored-by: Shaoshuai Shi <shaoshuaics@gmail.com>

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>

* add voxel roi pooling

* add voxel r-cnn head

* add voxel query

* add voxel r-cnn

* add Voxel R-CNN

* add infos about Voxel R-CNN

* add voxel_rcnn_car.yaml

* add infos about Voxel R-CNN
Co-authored-by: Shaoshuai Shi <shaoshuaics@gmail.com>

* Added CaDDN detector and support for image, depth map, and 2D GT box
dataloading

* Moved image flip augmentation to augmentor_utils

* Updated default get item list to include points

* Moved utils functions into transform_utils

* Combined FFE + F2V into ImageVFE, renamed FFE to FFN, moved depth downsample into data_processor

* Updated README with updated CaDDN weights

* Updated comments for image vfe

* support SECOND-IoU

* rename file

* modify readme and fix model name in __init__.py

* add codes for creating kitti infos and gt database