Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Custom model for image classification #180

Closed
neumartin opened this issue Dec 16, 2018 · 3 comments
Closed

Custom model for image classification #180

neumartin opened this issue Dec 16, 2018 · 3 comments

Comments

@neumartin
Copy link

In the sample DeepLearning_ImageClassification_TensorFlow I try to create my own model folowing this:
https://www.tensorflow.org/hub/tutorials/image_retraining

But when I execute the sample I get:

EXCEPTION
#########
Input column 'input' does not exist in the model
Parameter name: inputs

@neumartin
Copy link
Author

@CESARDELATORRE, @asthana86, @mariuszwojcik can you help me?

@CESARDELATORRE
Copy link
Contributor

Make sure the input tensor names match. You can open your TensorFlow model with a tool like Netron (https://github.com/lutzroeder/Netron), also coming in Visual Studio when you install "VS Tools for AI" to explore the TensorFlow model.

Check this intro: https://blogs.msdn.microsoft.com/dotnet/2018/09/12/announcing-ml-net-0-5/#added-a-tensorflow-model-scoring-transform-tensorflowtransform

Also, check this other ML.NET sample that I created:
https://github.com/CESARDELATORRE/TensorFlowImageClassificationWebAPI
In this case, the TensorFlow model is exported from Azure Cognitive Services Custom Vision, so it has different input names, etc.

@asthana86
Copy link
Contributor

Closing this issue, please feel free to open this if this doesn't solve the problem.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants