pytorch-discriminative-loss
pytorch-discriminative-loss copied to clipboard
Implementation of discriminative loss for instance segmentation by pytorch
pytorch-discriminative-loss
This repository implements Semantic Instance Segmentation with a Discriminative Loss Function by pytorch.
Pretrained model
- Download model from here and save it in model/
Environments
Ubuntu16.04
python3.6.4
pytorch: 0.3.1
opencv: 3.3.1
Reference
https://github.com/Wizaron/instance-segmentation-pytorch
https://github.com/milesial/Pytorch-UNet