Dicussion: TensorFlow for Image Analysis #241
Closed
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Start setting up TensorFlow for image analysis
GitHub issue(s):
#240
This PR is mainly to have a discussion of how to proceed with add neural network inferencing for images.
The code sets up a couple functions in Python to use tensorflow to do this following the example from: https://docs.databricks.com/_static/notebooks/tensorflow-and-spark.html
Ideal usage eventually should be
images.flatMap(apply_inference_on_batch)
whereimages
is the result ofExtractImageDetails
Couple issues I ran into:
get_tensor_by_name('softmax:0')
results in an errorKeyError: "The name 'softmax:0' refers to a Tensor which does not exist. The operation, 'softmax', does not exist in the graph."
I'm not yet familiar enough with TensorFlow. When I look this up online, a lot of people have they've encountered the same issue and said they have fixed it but did not post the solution.ExtractImageDetails
in pyspark would be an option too.Please suggest any ideas how to proceed.
Interested parties
@lintool @ruebot @ianmilligan1
Thanks in advance for your help with the Archives Unleashed Toolkit!