mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-23 21:11:08 +00: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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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 = np.expand_dims(test_image, axis=0)
result = classifier.predict(test_image)
training_set.class_indices
# training_set.class_indices
if result[0][0] == 0:
prediction = "Normal"
if result[0][0] == 1: