DeepSceneSeg / SSMA

TensorFlow model for training SSMA for multimodal semantic segmentation
http://deepscene.cs.uni-freiburg.de
GNU General Public License v3.0
85 stars 13 forks source link

Could you release data augmentation code? #3

Closed githubfragments closed 5 years ago

githubfragments commented 5 years ago

Thanks for your nice work. Could you release the methods about the data augmentation? As you referred in the paper, include rotatoin, skewing, scaling, vigetting, cropping, brightness modulation, contrast modulation, and flipping. And the methods of the disparity map converting to jet colorized three channel image, depth completion algorithm, and HHA. These methods are very helpful for studies.

avalada commented 5 years ago

The augmentation code is part of our internal TensorFlow but you can just use normal opencv functions. You can find the HHA computation code here https://github.com/s-gupta/rcnn-depth/blob/master/rcnn/saveHHA.m The code for depth completion is here https://github.com/kujason/ip_basic

githubfragments commented 5 years ago

Thanks very much. Another question: did the depth image performance the contrast and brightness modulation?

githubfragments commented 5 years ago

The augmentation code is part of our internal TensorFlow but you can just use normal opencv functions. You can find the HHA computation code here https://github.com/s-gupta/rcnn-depth/blob/master/rcnn/saveHHA.m The code for depth completion is here https://github.com/kujason/ip_basic

Did the depth images use the same data augmentation as the rgb images? @avalada

avalada commented 5 years ago

We did not use vignetting, brightness and contrast modulation for the depth images.

githubfragments commented 5 years ago

Thanks for your replies!