Segmentation to 3D postion ##Train commands: For big dataset: python3 ./train.py -d 'big' -b 64 -j8 For small dataset: python3 ./train.py -d 'small' -b 64 -j8