Skip to content

A PyTorch implementation of PointRend: Image Segmentation as Rendering

Notifications You must be signed in to change notification settings

zsef123/PointRend-PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PointRend

A PyTorch implementation of PointRend: Image Segmentation as Rendering

title


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 test_point_sampling.ipynb

Original Figure

fig5

Reference : Pytorch Deeplab Tutorial


How to use:

First, fix data path in default.yaml

Multi GPU Training See details in Single GPU Training

➜ python3 -m torch.distributed.launch --nproc_per_node={your_gpus} main.py -h

Sinle GPU Training

➜ python3 main.py -h
usage: main.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 main.py config/default.yaml test_codes

About

A PyTorch implementation of PointRend: Image Segmentation as Rendering

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published