Skip to content

PotatoTian/recall-semseg

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Recall Loss for Semantic Segmentation (This repo implements the paper: Recall Loss for Semantic Segmentation)

License: MIT

Overview

Update

The paper is accpeted to ICRA2022.

Download Synthia dataset

  • The model uses the SEQS-05 Sythia-seq collections here
  • Using different collections might require modifications to the dataloader. Please check the specific data structure and labels.
  • Extract the zip / tar and modify the path appropriately in your ./configs/deeplab_synthia.yml
path: /datasets/synthia-seq/

Create conda environment

conda env create -f requirements.yaml
conda activate recallCE

Training

  • Run the training script.
python train.py --config ./configs/deeplab_synthia.yaml

Testing

  1. For testing on Sythia dataset, change the validation split to test.
val:
    split: test
  1. Run the testing script
  • Qulitative results will be saved in runs/synthia/rgbd_synthia
python test.py --config ./configs/deeplab_synthia.yaml

Acknowledgments

  • This work was supported by ONR grant N00014-18-1-2829.
  • This code is built upon the implementation from Pytorch-semseg.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages