Yes. I am doing the transformation that the Caffe version does in it’s ‘Image Seg Data’ layer. Without any external data augmentation by say rotation etc, the Caffe version is able to reap the said performance. Hence, I have restricted myself to the same transforms as well.