diff --git a/ACL_TensorFlow/built-in/cv/MobileNetv1_for_ACL/scripts/mobilenetv1_preprocessing.py b/ACL_TensorFlow/built-in/cv/MobileNetv1_for_ACL/scripts/mobilenetv1_preprocessing.py index 1399b4dd8f6e73729fc5f73e50ff9f76c00444e8..0a2b3d578a4f401a4b5ebdd222cb2488d2ae3c66 100644 --- a/ACL_TensorFlow/built-in/cv/MobileNetv1_for_ACL/scripts/mobilenetv1_preprocessing.py +++ b/ACL_TensorFlow/built-in/cv/MobileNetv1_for_ACL/scripts/mobilenetv1_preprocessing.py @@ -382,7 +382,8 @@ def preprocess(src_path, save_path): if not os.path.isdir(file): print(file) img = convert_RGB(os.path.join(src_path, file)) - print(type(img)) + img = np.array(img) + img = tf.convert_to_tensor(img) img = preprocess_image(img, 224, 224,