docs: added resize to fix keras shape error (#174)

jina-ai · Oct 27, 2021 · 7e9c04f · 7e9c04f
1 parent 489014e
commit 7e9c04f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/get-started/celeba.md b/docs/get-started/celeba.md
@@ -32,6 +32,7 @@ def data_gen():
     for d in from_files('/Users/jina/Downloads/img_align_celeba/*.jpg', size=100, to_dataturi=True):
         d.convert_image_uri_to_blob()
         d.normalize_image_blob()
+        d.resize_image_blob(width=224, height=224) #: no need of this line if you are using pytorch/paddle
         d.set_image_blob_channel_axis(-1, 0)  #: no need of this line if you are using tf/keras
         yield d
 ```