Skip to content

Latest commit

 

History

History
17 lines (13 loc) · 581 Bytes

README.md

File metadata and controls

17 lines (13 loc) · 581 Bytes

Features

The script convert_param.py can be used to convert the pretrained model weights from PyTorch to PaddlePaddle.

The script visualization.py can render an image with its corresponding labels (either predicted results or groundtruth labels) for semantic visualization. Example is shown below. The color bar on the right hand side indicates the colors used for corresponding categories. vis_example

Usage:

python utils/visualization.py --img-dir path/to/image --label-dir path/to/label

More arguments can be found in the script.