pytorch-discriminative-loss icon indicating copy to clipboard operation
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.

Result

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