You can deploy this model and start earning money today!

ranvir-not-found/
resnet50-sda_gunshot-detection

Name: resnet50-sda_gunshot-detection
Availability: InStock

Views

No views yet

Fine-tuned ResNet-50 Model for Gunshot Detection

This is a fine-tuned ResNet-50 model for detecting gunshots in audio recordings. The model was trained on a dataset of audio clips labeled as either "gunshot" or "background".

Model Details

Fine-tuned by: Ranabir Saha
Fine-tuned on: Tropical forest gunshot classification training audio dataset from Automated detection of gunshots in tropical forests using convolutional neural networks (Katsis et al. 2022)
Dataset Source: https://doi.org/10.17632/x48cwz364j.3
Input: Mel-spectrograms (128x128x3) loaded from .npy files, generated from audio clips
Output: Binary classification (Gunshot/Background)

Training

The model was trained using the following parameters:

Base Model: ResNet-50 pre-trained on ImageNet
Optimizer: Adam (initial learning rate=0.0001, fine-tuning learning rate=0.00001)
Loss Function: Categorical cross-entropy
Metrics: Accuracy, Precision, Recall
Batch Size: 32
Initial Training: Up to 200 epochs with early stopping (patience=25) on validation loss
Fine-tuning: Last 10 layers unfrozen, up to 60 epochs with early stopping (patience=25)
Class Weights: Balanced to handle class imbalance

Usage

To use this model for inference, you can load it from the Hugging Face Hub and pass preprocessed mel-spectrograms loaded from .npy files as input.

Example

python

import numpy as np
import tensorflow as tf
from huggingface_hub import hf_hub_download

# Download the model
model_path = hf_hub_download(repo_id="ranvir-not-found/resnet50-sda_gunshot-detection", filename="resnet50_model.keras")
model = tf.keras.models.load_model(model_path)

# Function to load and preprocess .npy file
def load_and_preprocess_npy(file_path):
    # Load .npy file
    mel_spectrogram = np.load(file_path)
    # Ensure 3D shape: (height, width, channels)
    if mel_spectrogram.ndim == 2:
        mel_spectrogram = np.expand_dims(mel_spectrogram, axis=-1)
    # Resize to 128x128 if needed
    if mel_spectrogram.shape[:2] != (128, 128):
        mel_spectrogram = tf.image.resize(mel_spectrogram, (128, 128)).numpy()
    # Convert to 3-channel (RGB)
    mel_spectrogram = tf.repeat(mel_spectrogram, 3, axis=-1)
    # Apply ResNet-50 preprocessing
    mel_spectrogram = tf.keras.applications.resnet50.preprocess_input(mel_spectrogram)
    return mel_spectrogram

# Example usage
npy_path = "path/to/your/spectrogram.npy"
input_data = load_and_preprocess_npy(npy_path)
input_data = tf.expand_dims(input_data, axis=0)  # Add batch dimension
predictions = model.predict(input_data)
class_names = ['gunshot', 'background']
predicted_class = class_names[np.argmax(predictions[0])]
print(f"Predicted class: {predicted_class}, Probabilities: {predictions[0]}")

Evaluation

The model was evaluated on a validation set, and the following metrics were computed:

Accuracy
Precision
Recall
Classification Report
Confusion Matrix The model was optimized for recall on the 'gunshot' class.

For more details, please refer to the training script and logs.