/ Machine Learning

A PyTorch implementation of PointRend: Image Segmentation as Rendering

A PyTorch implementation of PointRend: Image Segmentation as Rendering

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

sample

dog

Original Figure

A-PyTorch-implementation

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

GitHub

Comments