Giter VIP home page Giter VIP logo

polysnake's Introduction

PolySnake

This is a PyTorch/GPU re-implementation of the paper Recurrent Contour-based Instance Segmentation with Progressive Learning city image

Any questions or discussions are welcomed!

Installation

Please see INSTALL.md.

Testing

Testing on Cityscapes

  1. Download the pretrained model from Google Drive or Baidu Cloud, and put it to $ROOT/data/model/rcnn_snake/cityscapes/.
  2. Test:
    # use coco evaluator
    python run.py --type evaluate --cfg_file configs/city_snake.yaml
    # use the cityscapes official evaluator
    python run.py --type evaluate --cfg_file configs/city_snake.yaml test.dataset CityscapesVal
    
  3. Speed:
    python run.py --type network --cfg_file configs/city_snake.yaml
    

Testing on Kins

  1. Download the pretrained model from Google Drive or Baidu Cloud, and put it to $ROOT/data/model/snake/kins/.
  2. Test:
    python run.py --type evaluate --cfg_file configs/kins_snake.yaml test.dataset KinsVal
    
  3. Speed:
    python run.py --type network --cfg_file configs/kins_snake.yaml test.dataset KinsVal
    

Testing on Sbd

  1. Download the pretrained model from Google Drive or Baidu Cloud, and put it to $ROOT/data/model/snake/sbd/.
  2. Test:
    python run.py --type evaluate --cfg_file configs/sbd_snake.yaml test.dataset SbdVal
    
  3. Speed:
    python run.py --type network --cfg_file configs/sbd_snake.yaml test.dataset SbdVal
    

Testing on COCO

  1. Download the pretrained model from Google Drive or Baidu Cloud, and put it to $ROOT/data/model/snake/coco/.
  2. Test:
    python run.py --type evaluate --cfg_file configs/coco_snake.yaml test.dataset CocoVal
    
  3. Speed:
    python run.py --type network --cfg_file configs/coco_snake.yaml test.dataset CocoVal
    

Demo

We support demo for image and image folder using python run.py --type demo --cfg_file /path/to/yaml_file demo_path /path/to/image ct_score 0.3.

For example:

python run.py --type demo --cfg_file configs/sbd_snake.yaml demo_path demo_images ct_score 0.3
# or
python run.py --type demo --cfg_file configs/city_snake.yaml demo_path demo_images/munster_000048_000019_leftImg8bit.png ct_score 0.3

If setup correctly, the output will be saved at $ROOT/demo_out/ and look like

demo

Training

The training parameters can be found in project_structure.md.

Training on Cityscapes

python train_net.py --cfg_file configs/city_snake.yaml model rcnn_snake det_model rcnn_det

Training on Kins

python train_net.py --cfg_file configs/kins_snake.yaml model kins_snake

Training on Sbd

python train_net.py --cfg_file configs/sbd_snake.yaml model sbd_snake

Training on COCO

python train_net.py --cfg_file configs/coco_snake.yaml model coco_snake

Citation

If you find this code useful for your research, please use the following BibTeX entry.

@inproceedings{PolySnake2023,
  title={Recurrent Contour-based Instance Segmentation with Progressive Learning},
  author={Feng, Hao and Zhou, Wengang and Yin, Yufei and Deng, Jiajun and Sun, Qi and Li, Houqiang},
  booktitle={Arxiv},
  year={2023}
}

Acknowledgement

Our work benefits a lot from DeepSnake and E2EC. Thanks for their wonderful works.

polysnake's People

Contributors

fh2019ustc avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.