Siamese network based Deep Co-segmentation implemented in PyTorch framework. Model architecture inspired by SegNet.

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

MIT

Sayan Goswami