update for getting original image while training
Merge branch 'jiajun_research' of https://github.com/amitgroup/parts-net into jiajun_research