caffe-segnet

Caffe SegNet

这是一个修改过的 Caffe 版本,支持 SegNet 架构 如 SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation Vijay Badrinarayanan, Alex Kendall 和 Roberto Cipolla, PAMI 2017 [http://arxiv.org/abs/1511.00561] 中所述

更新版本:

此版本支持 cudnn v2 加速。@TimoSaemann 有一个分支支持较新版本的 Caffe（2016 年 12 月），其中包含 cudnn v5.1： https://github.com/TimoSaemann/caffe-segnet-cudnn5

使用示例模型和网络摄像头演示开始

如果您只想尝试一个预训练的示例模型，则可以在以下链接中找到 SegNet webdemo 中使用的模型以及运行实时网络摄像头演示的脚本：https://github.com/alexgkendall/SegNet-Tutorial

示例网络规范和求解器原型文件位于 examples/segnet 中。要训练模型,请在 data 层中修改数据路径,使其对应于您的 dataset.txt 文件(如上所述)。在最后一个卷积层中,将 num_output 更改为您的数据集中的类数。