Pytorch Implementation of Predictive Models of Fire via Deep learning Exploiting Colorific Variation
-
Pytorch Implementaion of Nielsen-net with LSTM
-
Train example with randomly generate data
(Test and Validation are unavailable only for training)
- This implementaion does not contain Dataset
- Python 3.6 +
- Pytorch
- Opencv2
- Numpy
- tensorboardX
- argparse
pip install -r requirements.txt
To train with Datset:
$ python main.py --data=/custom/dataset/dir --label=/custom/dataset/label/dir --logdir=/path/to/logs
To train with Randomly Generated data:
$ python main.py
-
Thumbnails of Firedata
-
Thumbnails of Non-Firedata
