Skip to content

Data preparation and model training for the NOAA Puget Sound Nearshore Fish dataset

Notifications You must be signed in to change notification settings

agentmorris/noaa-fish

Repository files navigation

NOAA Puget Sound Nearshore Fish data curation code

Overview

The code in this folder prepares data for the NOAA Puget Sound Nearshore Fish dataset on lila.science, and trains a preliminary detector.

Gratuitous image with ground truth and predicted bounding boxes, from the results on the validation locations:

Files

prepare-noaa-fish-data-for-lila.py

Given the Excel file provided by NOAA, prepare data in COCO format for release on LILA.

prepare-noaa-fish-data-for-training.py

  • Convert annotations from COCO format to YOLO format
  • Split locations into train/val
  • Train YOLOv5 (command-line instructions, not done programmatically)
  • Validate and preview training results on train/val data

About

Data preparation and model training for the NOAA Puget Sound Nearshore Fish dataset

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages