InceptionV3 和使用 tensorflow 进行迁移学习 [英] InceptionV3 and transfer learning with tensorflow

问题描述

我想从 tensorflow 示例中给定的 inceptionV3 进行迁移学习.遵循分类图像示例以及此处给出的运算符和张量名称 https://github.com/AKSHAYUBHAT/VisualSearchServer/blob/master/notebooks/notebook_network.ipynb 我可以创建我的图表.但是，当我将一批大小为 (100, 299, 299, 3) 的图像放入预先计算的初始图中时，在 pool_3 层出现以下形状错误:

I would like to do a transfer learning from the given inceptionV3 in tensorflow example. Following the classify image example and the operator and tensor names given here https://github.com/AKSHAYUBHAT/VisualSearchServer/blob/master/notebooks/notebook_network.ipynb I can create my graph. But when, I put a batch of images of size (100, 299, 299, 3) in the pre-computed inception graph, I get the following shape error at the pool_3 layer :

ValueError: Cannot reshape a tensor with 204800 elements to shape [1, 2048] (2048 elements)

这个 inceptionV3 图似乎不接受图像批处理作为输入.我错了吗?

It seems that this inceptionV3 graph doesn't accept image batch as input. am I wrong ?

推荐答案

你没有错.这似乎是一个非常合理的功能请求，所以我在 github 上打开了一张票.按照那个更新.

You're not wrong. This seems like a very reasonable feature request, so I've opened a ticket for it on github. Follow that for updates.

这篇关于InceptionV3 和使用 tensorflow 进行迁移学习的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！