algorithm/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-07-05 17:34:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update cnn_classification.py (#8570)

Browse Source
This commit is contained in:
Rohan Anand
2023-03-30 10:39:21 +05:30
committed by GitHub
parent ac111ee463
commit a71f22dae5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
computer_vision/cnn_classification.py
View File

@ -93,7 +93,7 @@ if __name__ == "__main__":
test_image = tf.keras.preprocessing.image.img_to_array(test_image) test_image = tf.keras.preprocessing.image.img_to_array(test_image)
test_image = np.expand_dims(test_image, axis=0) test_image = np.expand_dims(test_image, axis=0)
result = classifier.predict(test_image) result = classifier.predict(test_image)
training_set.class_indices # training_set.class_indices
if result[0][0] == 0: if result[0][0] == 0:
prediction = "Normal" prediction = "Normal"
if result[0][0] == 1: if result[0][0] == 1: