Python Apply Portorch For Segmentation Code In Python