Batch Instance Segmentation Pytorch