Skip to content

Releases: UPstartDeveloper/Fire-Detection-API

Fire Classifier (CNNv2)

16 Sep 19:46
Compare
Choose a tag to compare

This model uses an architecture that is nearly identical to that of the previous release, except we now use more pooling layers and adjusted the numbers of filters and nodes to make it more memory efficient.

The entire model code is at the bottom of the notebook here. It accomplished a validation accuracy of about 83% (better than the previous CNN model).

Fire Classifier (CNNv1)

15 Sep 21:50
Compare
Choose a tag to compare
  • This model is the first iteration of using convolutional neural networks to solve this problem.
  • This is the first model which has been trained on a balanced dataset of fire/normal images.
  • The entire model code is at the bottom of the notebook here.

Fire Classifier Model (MLPv2)

31 Aug 20:16
Compare
Choose a tag to compare
  • This model fixes an error in the last layer, replacing the incorrectly used softmax activation function with sigmoid
  • Reduces memory size by saving only the model weights and biases (aka "parameters") to an H5 file, and the information about the model architecture (i.e. the "layers" of the neural network) to JSON

Fire Classifier Model (MLPv1)

26 Aug 23:07
Compare
Choose a tag to compare

This is a MLP model trained on the image dataset here.

The notebook is here.