Caffe:如何训练端到端(图像到图像)?

Caffe: How to train end-to-end (image to image)?

我们对 caffe 还很陌生,但到目前为止我们所看到的看起来很有希望。

读了几篇论文后(1,2), we wanted to reproduce the result of 1, specifically about a segmentation challenge 4.

我们从3 and were able to execute it, just to see, that the trained network didn't work with the dataset from 4下载了修改后的caffe。

起初我们认为网络需要针对特定​​问题进行训练。 这导致了如何做的问题 'image-to-image (aka end-to-end) learning ' (4, training data).

这将我们引向 'holistically nested edge detection'(hed,2),其中似乎使用了图像到图像学习。 借助 hed,我们能够自行重新训练网络。但它不起作用(如果我们尝试为 4 的数据集训练网络,它会导致所有 0 或 0.5 个图像 - 黑色图像 :-( )。为了初始化,我们编写了一个脚本来计算均值- map 我们用于 4.

的数据集

我们的问题是:

亲切的问候, 克劳斯和伯恩哈德

Ps:我们在 caffe-user 组问了同样的问题,并打算 post 两个 locations 的解决方案。

一段时间后,尝试了几种不同的方法 - 我偶然发现:

https://github.com/naibaf7

使用那个 caffe fork,caffe_neural_models 和 caffe_neural_tool 训练图像(原始)到图像(标签)可以非常简单地完成。

只需查看 'caffe_neural_models/net*' 了解不同的配置。