如何在预训练的VGG16上训练FCN模型

时间:2019-03-07 14:02:31

标签: python tensorflow pre-trained-model semantic-segmentation

我是语义分割的新手,我实现了FCN网络,现在我想尽量不要从头开始训练,而要使用训练前的vgg16权重。我看到了类似link的实现,但是我不确定新的数据集输入将进入网络。 更清楚地说,在上面的链接中,vvg部分返回来自经过训练的网络的输入图像以及输出层3、4、7。

 image_input, pool3_out, pool4_out, fc7_out = self._load_vgg16()

我不确定新一​​批数据进入模型的位置。感谢您的指导。

0 个答案:

没有答案