rujutashashikanjoshi's picture
Upload README.md with huggingface_hub
aafd670 verified
|
Raw
History Blame
1.12 kB
metadata
license: agpl-3.0
tags:
  - object-detection
  - YOLO26
  - computer-vision
  - ultralytics
library_name: ultralytics
pipeline_tag: object-detection

YOLO26 MEDIUM Model

Fine-tuned YOLO26 model for object detection.

Model Details

  • Architecture: YOLO26Medium
  • Framework: Ultralytics YOLO26
  • Resolution: 640x640
  • Epochs: 100
  • Batch Size: 16

Classes

car

Usage

from ultralytics import YOLO
from huggingface_hub import hf_hub_download

model_path = hf_hub_download(repo_id="rujutashashikanjoshi/yolo26-vehicle-detection-4139_full-v2_1-100m", filename="best.pt")
model = YOLO(model_path)

results = model("your_image.jpg", conf=0.25)
results[0].show()

Training Config

{
    "model_type": "medium",
    "epochs": 100,
    "batch_size": 16,
    "image_size": 640,
    "learning_rate": 0.01,
    "weight_decay": 0.0005,
    "momentum": 0.937,
    "warmup_epochs": 3,
    "workers": 4,
    "patience": 30,
    "save_period": 10,
    "conf_threshold": 0.25,
    "iou_threshold": 0.45,
    "output_dir": "./output/4139_v2_1-100m"
}

License

AGPL-3.0