Skip to content

Latest commit

 

History

History
35 lines (25 loc) · 1.12 KB

README.md

File metadata and controls

35 lines (25 loc) · 1.12 KB

*** NOTE: This code is deprecated and being slowly merged to ami-ml repoistory. ***

Species model trainer from GBIF

This repository contains the code to download image and metadata for a list of species from Global Biodiversity Information Facility (GBIF) and train a deep learning model using the downloaded data.

Setup python environment

Install conda and prepare a python environment using the following steps:

Build a new conda environment

conda create -n gbif_species_trainer python=3.9

Activate the conda environment:

conda activate gbif_species_trainer

Install cuda toolkit and pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

Install additional libraries using pip:

python3 -m pip install -r requirements.txt

Download Data

Follow instructions in the data_download folder to download data for a list of species.

Train Model

Train a model using scripts in model_training folder.