ogaith's picture
Add YOLOv8 dog detector weights, README, inference example and assets
826ddf5
|
Raw
History Blame Contribute Delete
1.21 kB
metadata
license: apache-2.0
base_model: Ultralytics/YOLOv8
tags:
  - ultralytics
  - object-detection
  - yolo
  - yolov8
  - dog
pipeline_tag: object-detection
library_name: ultralytics

🐶 YOLOv8 Dog Detector

This repository contains a finetuned YOLOv8 model for detecting dogs in images.

  • Model: best.pt
  • Dataset: Google Open Images v7 (~4,000 dog instances)
  • Task: Object Detection (1 class: dog)

📦 Installation

pip install ultralytics

🚀 CLI Usage (recommended for quick testing)

yolo detect predict \
    model=best.pt \
    source="assets/runs/detect/val/example1.jpg" \
    imgsz=640 \
    conf=0.25 \
    save=True

Batch infer on folder

yolo detect predict model=best.pt source="images_folder/" imgsz=640 conf=0.25 save=True

🐍 Python Usage

from ultralytics import YOLO

# Load model
model = YOLO("best.pt")

# Run prediction
results = model.predict(
    source="assets/runs/detect/val/example1.jpg", 
    conf=0.25, 
    imgsz=640
)

# Display or save result
results[0].show()   # display
# results[0].save() # save to file