ICICLE-AI/CAN_Benchmark
Viewer β’ Updated β’ 1.09k β’ 15
How to use ICICLE-AI/yolov9-animals-AE-data with ultralytics:
# Couldn't find a valid YOLO version tag.
# Replace XX with the correct version.
from ultralytics import YOLOvXX
model = YOLOvXX.from_pretrained("ICICLE-AI/yolov9-animals-AE-data")
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)This model is a YOLOv9 detector fine-tuned with Ultralytics.
It was trained for 50 epochs on a subset of the ICICLE-AI/CAN_Benchmark dataset containing three species:
args.yamlThe model converged by ~40 epochs, and shows strong precision/recall on the held-out validation set:
Load the model directly from the Hugging Face Hub:
from ultralytics import YOLO
# Load model from HF Hub
model = YOLO("ICICLE-AI/yolov9-animals-AE-data")
# Run inference
results = model("demo.jpg")
results[0].show()