DeepSeaAI is a Python package to simplify processing deep sea video in AWS from a command line.
Project description
DeepSeaAI is a Python package to simplify processing deep sea video in AWS from a command line.
It includes reasonable defaults that have been optimized for deep sea video. The goal is to simplify running these algorithms in AWS.
DeepSea-AI currently supports:
- Training YOLOv5 object detection models
- Processing video with YOLOv5 detection and tracking pipelines using either:
- DeepSort tracking
- StrongSort tracking
Install
Setup an AWS account.
After you have setup your AWS account, configure it using the awscli tool, and confirm your AWS Account by listing your s3 buckets
pip install awscli
aws configure
aws --version
aws s3 ls
Install and update using pip in a Python>=3.8.0 environment:
pip install -U deepsea-ai
Setup your AWS account for use with this module with
deepsea-ai setup
Tutorials
- FathomNet ✨ Recommended first step to learn more about how to train a YOLOv5 object detection model using freely available FathomNet data
The best way to use the tutorials is with Anaconda.
Create the Anaconda environment
This will create an environment called deepsea-ai-notebooks and make that available in your local jupyter notebook as the kernel named deepsea-ai-notebooks
conda env create
conda activate deepsea-ai-notebooks
pip install ipykernel
python -m ipykernel install --user --name=deepsea-ai-notebooks
Launch jupyter
cd docs/notebooks
jupyter notebook
Commands
deepsea-ai setup --help
- Setup the AWS environment. Must run this once before any other commands.deepsea-ai train --help
- Train a YOLOv5 model and save the model to a bucketdeepsea-ai process --help
- Process one or more videos and save the results to a bucketdeepsea-ai ecsprocess --help
- Process one or more videos using the Elastic Container Service and save the results to a bucketdeepsea-ai split --help
- Split your training data. This is required before the train command.deepsea-ai -h
- Print help message and exit.
Source code is available at github.com/mbari-org/deepsea-ai.
For more details, see the official documentation.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for deepsea_ai-1.15.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3bb536cf636100448dbd75eb52bc9bad81193e7a7cd75ba7b9da4fe8919b2f8d |
|
MD5 | 12d8f96d6a3c396de85cc1a68ea8ddbd |
|
BLAKE2b-256 | b6de4cc49639c0726c724aa6431ad8b2395bd8772bca1c815e075d25cf98af44 |