Upload 16 files

.gitignore

@@ -0,0 +1,20 @@
+# Compiled Python files
+*.pyc
+__pycache__/
+
+# Virtual environment
+venv/
+ENV/
+
+# Jupyter Notebook checkpoints
+.ipynb_checkpoints/
+
+# IDE settings
+.vscode/
+.idea/
+
+# Python cache files
+*.pyc
+
+# Gradio cache files
+.gradio/

app.py

@@ -0,0 +1,42 @@
+import gradio as gr
+import numpy as np
+import cv2
+import tensorflow as tf
+
+# Load the model
+vgg16_model = tf.keras.models.load_model('fish_vgg16_model.h5')
+mobilenet_model = tf.keras.models.load_model('fish_mobilenet_model.h5')
+custom_cnn_model = tf.keras.models.load_model('fish_cnn_model.h5')
+
+# Define the preprocessing function
+def preprocess_image(image):
+    image = cv2.resize(image, (224, 224))
+    image = image / 255.0
+    return image
+
+# Define the prediction function
+def predict(image):
+    # Preprocess the image
+    processed_image = preprocess_image(image)
+    
+    # Make predictions using the models
+    vgg_pred = vgg16_model.predict(np.expand_dims(processed_image, axis=0))[0]
+    mobilenet_pred = mobilenet_model.predict(np.expand_dims(processed_image, axis=0))[0]
+    custom_cnn_pred = custom_cnn_model.predict(np.expand_dims(processed_image, axis=0))[0]
+    
+    # Get the predicted labels
+    vgg_label = np.argmax(vgg_pred)
+    mobilenet_label = np.argmax(mobilenet_pred)
+    custom_cnn_label = np.argmax(custom_cnn_pred)
+    
+    return {
+        "VGG16 Prediction": vgg_label,
+        "MobileNet Prediction": mobilenet_label,
+        "Custom CNN Prediction": custom_cnn_label
+    }
+
+# Create the Gradio interface
+inputs = gr.inputs.Image()
+outputs = gr.outputs.Label(num_top_classes=3)
+
+gr.Interface(fn=predict, inputs=inputs, outputs=outputs).launch()

fish_cnn_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ead2b2b8e6f943d65c9b408ee435f3fb4a1d587ff65fed9f5b9ec7d2f7b4dd3
+size 116259088

fish_mobilenet_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e25eb25c5e5ad3c261c0097f48e08e3fafef7bcd56a6bd4462bb8459d42cdcc
+size 19440064

fish_vgg16_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9cb8debb8d6a0e49336daac4c85a9df87412024dc57843829dfee9877f186a48
+size 213127872

requirements.txt

@@ -0,0 +1,4 @@
+gradio==2.3.6
+opencv-python==4.5.3.56
+numpy==1.19.5
+tensorflow==2.6.0