PointRend的PyTorch实现:图像分割作为渲染
A PyTorch implementation of PointRend: Image Segmentation as Rendering
Python 机器学习
共203Star
详细介绍
PointRend
A PyTorch implementation of PointRend: Image Segmentation as Rendering
[arxiv] [Official Implementation: Detectron2]
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.
Original Figure
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
-
458 Star
-
2881 Star
-
502 Star
-
320 Star
-
1939 Star
-
130 Star
-
0 Star
-
0 Star
-
0 Star
-
416 Star