Update 06_classify_image.py

correction of classify
jensleitloff · Mar 25, 2019 · e956484 · e956484 · liaduarte · Mar 30, 2020
1 parent 39e2d10
commit e956484
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/py/06_classify_image.py b/py/06_classify_image.py
@@ -40,6 +40,17 @@
 in_rows_half = int(input_rows/2)
 in_cols_half = int(input_cols/2)
 
+# import correct preprocessing
+if input_channels is 3:
+    from image_functions import preprocessing_image_rgb as preprocessing_image
+else:
+    from image_functions import preprocessing_image_ms as preprocessing_image
+
+# pad image
+image = pad(image, ((input_rows, input_rows),
+                    (input_cols, input_cols),
+                    (0, 0)), 'symmetric')
+
 # don't forget to preprocess
 image = preprocessing_image(image)
 num_rows, num_cols, _ = image.shape