norbertm/jwst-quality-analysis-dataset
Viewer • Updated • 2.71k • 11
How to use norbertm/jwst-astronomical-detection with ultralytics:
from huggingface_hub import hf_hub_download
from ultralytics import YOLO
# pick the weights file from this repo's "Files and versions" tab
weights = hf_hub_download("norbertm/jwst-astronomical-detection", "<weights>.pt")
model = YOLO(weights)
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.