/pytorch-discriminative-loss

Implementation of discriminative loss for instance segmentation by pytorch

Primary LanguageJupyter Notebook

Implementation of Semantic Instance Segmentation with a Discriminative Loss Function paper using Pytorch

This repository this paper: Semantic Instance Segmentation with a Discriminative Loss Function.

Result

Currently implementing improvements over the original code

  • Generic instance number support
  • Fixed loss function implementation
  • Support for CVPPP 2017 LSC dataset

Reference

https://github.com/Wizaron/instance-segmentation-pytorch
https://github.com/milesial/Pytorch-UNet
https://github.com/davyneven/fastSceneUnderstanding