ziaddBou/pneumodoc-model

Model Description

This model employs a MobileNetV3 architecture fine-tuned for the detection of pneumonia from chest X-ray images. It is designed to assist radiologists by providing a preliminary automated diagnosis. tetststtststststtstst

Training Data

The model was trained on the Kaggle Pneumonia dataset, which contains thousands of labeled chest X-ray images from children.

Model Architecture

The model uses MobileNetV3 as the base for feature extraction, with additional custom layers to tailor it for pneumonia detection.

Training Procedure

The model was trained with an Adam optimizer and early stopping based on validation loss to prevent overfitting. Data augmentation techniques such as rotations and flips were used to enhance generalization.

Performance

The model achieved a high accuracy on the validation set, with the following metrics:

• Accuracy: XX%
• Precision: XX%
• Recall: XX%
• F1 Score: XX%

Usage

Here is an example of how to use this model:

import gradio as gr
import tensorflow as tf

model = tf.keras.models.load_model('model.h5')

def predict(image):
    processed_image = preprocess_image(image)
    return model.predict(processed_image)

iface = gr.Interface(fn=predict, inputs="image", outputs="label")
iface.launch()