An open API service indexing awesome lists of open source software.

https://github.com/say4n/deepcosegmentation.pytorch

Siamese network based Deep Co-segmentation
https://github.com/say4n/deepcosegmentation.pytorch

co-segmentation deep-learning image-segmentation neural-network pytorch pytorch-cnn pytorch-implmention

Last synced: 10 months ago
JSON representation

Siamese network based Deep Co-segmentation

Awesome Lists containing this project

README

          

# Deep Co-segmentation [PyTorch]

Siamese network based Deep Co-segmentation implemented in PyTorch framework.
Model architecture inspired by [SegNet](http://mi.eng.cam.ac.uk/projects/segnet/).

## files

`src/model.py` - pytorch model

`src/train.py` - train model from scratch

`src/inference.py` - infer results from trained model

`src/dataset.py` - dataloader for iCoseg dataset

## license

MIT

## author

Sayan Goswami