Lalitadityaa/CustomCNN_forest_fire

Model Card for ForestFireCNN Forest Fire Detection

This repository contains the PyTorch state dictionary (`state_dict`) for a custom ForestFireCNN model trained to detect forest fires from ground-level images.

Model Details

Model Description

Developed by: Lalitaditya M

Model type: PyTorch State Dictionary

Architecture: ForestFireCNN (Custom CNN)

Fine-tuned from model: N/A (Custom Architecture)

How to Get Started with the Model

This repository contains only the trained model weights (state_dict). Before loading the weights, recreate the ForestFireCNN architecture using the provided forestfirecnn.py file.

import torch
from forestfirecnn import ForestFireCNN

# Create the model architecture
model = ForestFireCNN(num_classes=2)

# Load the state dictionary
state_dict = torch.load(
    "lightweightcnn.pt",
    map_location="cpu"
)

model.load_state_dict(state_dict)
model.eval()

Repository Contents

lightweightcnn.pt
forestfirecnn.py
README.md