This is a project on A Fire Hazard Identification and Evaluation Framework for Fire Extinguishing Applications
Link to writeup and poste to be posted
- Results are in pipeline.ipynb
- See the two classifier.py files for functions of training, visualizing and data augmentation
- See utils.py for detailed helper functions
- Create a data folder in the home directory
- Create sub-folders with the name of dataset (i our case, data/FMD, data/fire_new)
- Under each sub-folder, have folders called /train and /val
- Under /train and /val, have folders with thei classes as folder names
- Place images in those class name folders