Classifying corals with Tensorflow
This repository uses transfer learning on the VGG19 model for the purpose of classifying coral species.
- With the current dataset, this model seems to overfit. This may be due to the lack of pictures in some species.
- Pictures MUST be cropped to ONLY have the coralites in frame. Originally, this was done manually, however looking forward there should be a solution in place to fix this labor intensive job.
For all scripts, the config settings are at the top of the file.
The file used to train the model. This file on first setup downloads VGG19, which should take a while. After that, it will access the cached VGG Model. This script requires you to have separate folders setup for your species, which can be done with script.py
You should probably run this with a beefy computer if you have 2000+ photos.
This file is used to test the trained model. It requires you to have validation data set up in folders. The script will then match the model accuracies with the validation data and plot them accordingly
This file is used to download the observations from a csv retrieved from iNaturalist. The input csv must be from iNaturalist since it reads and downloads based on the iNaturalist csv format.
This file is used to convert the species folders to genus/species folders.
This folder is the output for app3load.py The folder name must be manually changed before running, otherwise it will override the current folder.
This repo was set up by Collin Hurst: collinhurst@gmail.com