norbertm/jwst-quality-analysis-dataset
Viewer • Updated • 2.71k • 8
How to use norbertm/jwst-astronomical-detection with ultralytics:
# Couldn't find a valid YOLO version tag.
# Replace XX with the correct version.
from ultralytics import YOLOvXX
model = YOLOvXX.from_pretrained("norbertm/jwst-astronomical-detection")
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)This is a fine-tuned YOLO model specifically trained for detecting astronomical objects in JWST (James Webb Space Telescope) images.
from ultralytics import YOLO
# Load the model
model = YOLO("norbertm/jwst-astronomical-detection")
# Run inference
results = model("path/to/jwst/image.png", conf=0.15)
If you use this model in your research, please cite:
@dataset{jwst_quality_analysis,
title={JWST Quality Analysis Dataset},
author={Your Name},
year={2024},
url={https://huggingface.co/datasets/norbertm/jwst-quality-analysis-dataset}
}
MIT License - see LICENSE file for details.