skin_disease / README.md
SERHANI's picture
Upload README.md with huggingface_hub
a3ce54c verified
|
raw
history blame
429 Bytes

YOLO Custom Model

This is a YOLO model trained on custom data.

Model Description

  • Model Type: YOLO
  • Training Data: Custom Dataset
  • Input: Images
  • Output: Bounding boxes with class predictions

Usage

from ultralytics import YOLO

# Load the model
model = YOLO('model.pt')

# Make predictions
results = model('image.jpg')

Training Details

  • Framework: Ultralytics YOLOv8
  • Training Device: cuda