Skip to content

RashedDoha/meta-drn-pytorch

Repository files navigation

Meta-DRN: Meta-Learning for 1-Shot Image Segmentation


Description

PyTorch implementation of the paper Meta-DRN: Meta-Learning for 1-Shot Image Segmentation

How to train the model

First install dependencies

# clone project   
git clone https://github.com/RashedDoha/meta-drn-pytorch

# install project dependencies 
cd meta-drn-pytorch
pip install -r requirements.txt

Next navigate to train.py and run it to train the model

# training script

cd training

python train.py