Keras CV (#1270)

http://b/270144038

Author: Philmod

Dockerfile.tmpl

@@ -526,7 +526,8 @@ RUN pip install flashtext \
         accelerate==0.12.0 \
         catalyst \
         # b/206990323 osmx 1.1.2 requires numpy >= 1.21 which we don't want.
-        osmnx==1.1.1 && \
+        osmnx==1.1.1 \
+        keras-cv && \
     apt-get -y install libspatialindex-dev
 RUN pip install pytorch-ignite \
         qgrid \

tests/test_kerascv.py

@@ -0,0 +1,21 @@
+import unittest
+
+import json
+import keras_cv
+import keras
+import numpy as np
+
+class TestKerasCV(unittest.TestCase):
+    def test_inference(self):
+        classifier = keras_cv.models.ImageClassifier.from_preset(
+            'efficientnetv2_b0_imagenet_classifier',
+            load_weights=False, # load randomly initialized model from preset architecture with weights
+        )
+        image = keras.utils.load_img('/input/tests/data/face.jpg')
+        image = np.array(image)
+        keras_cv.visualization.plot_image_gallery(
+            [image], rows=1, cols=1, value_range=(0, 255), show=True, scale=4
+        )
+        predictions = classifier.predict(np.expand_dims(image, axis=0))
+        top_classes = predictions[0].argsort(axis=-1)
+        self.assertEqual(1000, len(top_classes))