fangzy97 / drnet

DRNet

This code is for the paper "DRNet: Double Recalibration Network for Few-Shot Semantic Segmentation". The implementation of the DRNet* is in drnetv2

Usage

Dependencies

• Python 2.7
• Pytorch 1.3.1

The dependencies can be installed by running:

pip install -r requirements.txt

Dataset

• Pascal-5ⁱ: VOC2012 + SBD
• COCO-20ⁱ: COCO2014

Models

We provide trained Models with the ResNet-50 backbone on Pascal-5ⁱ and COCO-20ⁱ for performance evalution. You can download from here.

Scripts

• Change the dataset path in db_path.py, ss_datalayer.py line 342 and train_frame_coco.py.
• Pascal-5ⁱ
  • Run the train scripts

    cd scripts
    ./train_groupX.sh
    

  • Run the evalution scripts for Pascal-5ⁱ

    python test_frame_all.py
    

• COCO-20ⁱ
  • Run the train scripts

    cd scripts
    ./coco_train_groupX.sh
    

  • Run the evalution scripts for Pascal-5ⁱ

    python test_frame_all_coco.py