Skip to content

Experiments in Identification of Individual Animals from Images using Deep Learning

Notifications You must be signed in to change notification settings

decotoj/animal_id

Repository files navigation

Example for setting up a data pipeline and training a neural net for classification

Utilizes structure from PyTorch tutorial here: https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

#################################

resize.py

Processes a folder of raw images (usually of relatively high resolution) organized in subfolders by class and resizes them and randomly assigns images to the train, validation or test sets

#################################

cow_ID_v1.py

Loads the training and validation data sets. Loads a pre-trained PyTorch model and trains it to perform image classification on the training data set. Model saved for re-use.

#################################

About

Experiments in Identification of Individual Animals from Images using Deep Learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages