Pyramid Scene Parsing Network.pdf
- 下载模型Google Drive，放入
model目录. - 准备图片，运行
RunnerOne.py即可:
Runner(is_flip=False, num_classes=19, log_dir="./model", save_dir="./output").run("data/input/test.png")
| Input image | Output image |
|---|---|
 |
 |
 |
 |
ReaderTrainImageAndLabel- 多线程读取训练的图片和标签
ReaderTestImage- 多线程读取测试的图片
Network- 定义网络的基本操作
PSPNet- 定义网络的具体结构
Tools- 工具类
RunnerTrain.py- 训练模型，输入是训练图片和标签的list，输出是训练好的模型
RunnerOne.py- 使用模型，输入是一张图片，输出是语义分割结果
RunnerAll.py- 使用模型，输入是图片的list，输出是语义分割结果