nyoki-mtl / pytorch-discriminative-loss

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