Skip to content

Commit b73a4a2

Browse files
authored
Added list of SUPPORTED_MODELS to README (#132)
1 parent a556c86 commit b73a4a2

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

README.md

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,13 @@ evaluator = MulticlassClassificationEvaluator(metricName="accuracy")
156156
print("Training set accuracy = " + str(evaluator.evaluate(predictionAndLabels)))
157157
```
158158

159+
`DeepImageFeaturizer` supports the following models from Keras:
160+
161+
* InceptionV3
162+
* Xception
163+
* ResNet50
164+
* VGG16
165+
* VGG19
159166

160167
### Distributed hyperparameter tuning
161168

@@ -238,6 +245,8 @@ predictor = DeepImagePredictor(inputCol="image", outputCol="predicted_labels", m
238245
predictions_df = predictor.transform(image_df)
239246
```
240247

248+
`DeepImagePredictor` supports the same set of models from Keras as `DeepImageFeaturizer`. (See above.)
249+
241250
##### For TensorFlow users
242251
Deep Learning Pipelines provides an MLlib Transformer that will apply the given TensorFlow Graph to a DataFrame containing a column of images (e.g. loaded using the utilities described in the previous section). Here is a very simple example of how a TensorFlow Graph can be used with the Transformer. In practice, the TensorFlow Graph will likely be restored from files before calling `TFImageTransformer`.
243252

python/sparkdl/transformers/named_image.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,9 @@
3131
from sparkdl.param import keyword_only, HasInputCol, HasOutputCol, SparkDLTypeConverters
3232
from sparkdl.transformers.tf_image import TFImageTransformer
3333

34+
35+
# If this list of supported models is expanded, update the list in the README
36+
# section for DeepImageFeaturizer.
3437
SUPPORTED_MODELS = ["InceptionV3", "Xception", "ResNet50", "VGG16", "VGG19"]
3538

3639

0 commit comments

Comments
 (0)