This model was trained to detect sea urchins using the YOLO11 architecture. Trained on open datasets to identify and locate urchins in various underwater conditions.
Below are the key metrics from the model evaluation on the validation set:
Training Validation Results
Training and Validation Losses
Training and Validation Losses
Confusion Matrix
Confusion Matrix
Precision-Recall Curve
Precision-Recall Curve
F1 Score Curve
F1 Score Curve
Training Configuration
Nvidia 4070 Geoforce RTX Super
Model Weights File: yolo11x_urchin_trained.pt
Number of Epochs: 100
Learning Rate: 0.001
Batch Size: 16
Image Size: 640x640
Deployment
How to Use the Model
To use the trained model, follow these steps:
Load the Model:
python
1from ultralytics import YOLO
23# Load the model4model = YOLO("yolo11x_urchin_trained.pt")
Limitations
The model was trained on a mix of open source images. It may not generalize well to other environments or non-marine scenarios. Additionally, environmental variations, occlusions, or poor lighting may affect performance.
Additional Notes:
Dataset Sources:
Two datasets were combined to improve model robustness, allowing the model to adapt to varying lighting and water conditions.
Ethical Considerations:
The detection results should be validated before using them for critical applications. The model’s performance in new environments might vary, and it may have biases if certain types of sea urchins were underrepresented in the training datasets.
Disclaimer
This repository is a scientific product and is not official communication of the National Oceanic and Atmospheric Administration, or the United States Department of Commerce. All NOAA project content is provided on an ‘as is’ basis and the user assumes responsibility for its use. Any claims against the Department of Commerce or Department of Commerce bureaus stemming from the use of this project will be governed by all applicable Federal law. Any reference to specific commercial products, processes, or services by service mark, trademark, manufacturer, or otherwise, does not constitute or imply their endorsement, recommendation or favoring by the Department of Commerce. The Department of Commerce seal and logo, or the seal and logo of a DOC bureau, shall not be used in any manner to imply endorsement of any commercial product or activity by DOC or the United States Government.
