use groudtruth shape for training