Skip to content

A PyTorch implementation of PointRend: Image Segmentation as Rendering

Notifications You must be signed in to change notification settings

fzhar/PointRend-PyTorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PointRend

A PyTorch implementation of PointRend: Image Segmentation as Rendering

title

[arxiv] [Official Repo] : Detectron2 (unopend now)


This repo for Only Semantic Segmentation on the PascalVOC dataset.

Many details differ from the paper for feasibilty check.


Reproduce Fig 5.

Sampled Points showing from different strategies on A Dog image.

See sample_semantic.ipynb

result

dog

Original Figure

fig5

Reference : Pytorch Deeplab Tutorial


How to use:

First, fix data path in default.yaml

➜ python3 train.py -h
usage: train.py [-h] config save

PyTorch Object Detection Training

positional arguments:
  config      It must be config/*.yaml
  save        Save path in out directory

optional arguments:
  -h, --help  show this help message and exit

e.g.)

python3 train.py config/default.yaml test_codes

About

A PyTorch implementation of PointRend: Image Segmentation as Rendering

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.3%
  • Python 0.7%