资源分类
》 资源 》算法》Semantic-Segmentation-PSPNet

Semantic-Segmentation-PSPNet

2020-01-17 | |  36 |   0 |   0

Paper

Pyramid Scene Parsing Network.pdf

Run Image

  1. 下载模型Google Drive，放入model 目录.

  2. 准备图片，运行RunnerOne.py 即可:
    Runner(is_flip=False, num_classes=19, log_dir="./model", save_dir="./output").run("data/input/test.png")

Run Result

图片.png

1. Data - ImageReader.py

  • ReaderTrainImageAndLabel

    • 多线程读取训练的图片和标签

  • ReaderTestImage

    • 多线程读取测试的图片

2. Model - PSPNet.py

  • Network

    • 定义网络的基本操作

  • PSPNet

    • 定义网络的具体结构

3. Tool - Tools.py

  • Tools

    • 工具类

4. Runner - RunnerXXX.py

  • RunnerTrain.py

    • 训练模型，输入是训练图片和标签的list，输出是训练好的模型

  • RunnerOne.py

    • 使用模型，输入是一张图片，输出是语义分割结果

  • RunnerAll.py

    • 使用模型，输入是图片的list，输出是语义分割结果

Reference


上一篇：pspnet-pytorch

下一篇：ERF-PSPNET

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...