search

databricks / spark-deep-learning

Deep Learning Pipelines for Apache Spark
https://databricks.github.io/spark-deep-learning
Apache License 2.0
1.99k stars 494 forks source link

Output multiple tensors from TFImageTransformer #179

Open gyscos opened 5 years ago

gyscos commented 5 years ago

Hi! From a dataframe of images, how could I run a detection model, and retrieve in one pass both the detection boxes and the detection classes?

I can currently use either outputTensor="detection_classes:0" or outputTensor="detection_boxes:0" when defining the TFImageTransformer, but I can't seem to output both, without running the transformer twice.

Is there a way to output multiple tensors from a single TFImageTransformer?