Deep Co-segmentation [PyTorch]

Siamese network based Deep Co-segmentation implemented in PyTorch framework. Model architecture inspired by 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