apply augmentation to dataset at training time

alr-internship / self-supervised-depth-denoising

Denoising YCB Objects with a self-supervised deep neural network

MIT License

9 stars 0 forks source link

apply augmentation to dataset at training time #8

Closed claudius-kienle closed 2 years ago

claudius-kienle commented 2 years ago

Applies augmentation to dataset, when the enable_augmentation flag is active. Augmentations are chosen randomly from a set of common ones. Package used is imgaug.

claudius-kienle commented 2 years ago

Update: Augmentation might generate invalid results, as RGBD datasets cant be resized/translated (otherwise point cloud invalid)