Class	Description
`wound`	Open injuries — lacerations, cuts, abrasions, stab wounds
`burn`	Thermal damage — 1st, 2nd, 3rd degree burns
`contusion`	Closed injuries — bruises, swelling without skin break

Training Details

Base model: YOLOv8n-cls (ImageNet pretrained)
Dataset: 3 Kaggle sources, quality-audited (blur, resolution, dedup), class-balanced via ambulance-specific augmentation
Optimizer: AdamW, lr=0.001, freeze=10 layers, dropout=0.3
Image size: 224×224

Usage

python

from ultralytics import YOLO

model = YOLO("best.pt")
results = model.predict("wound_image.jpg")
print(results[0].names[results[0].probs.top1])

Intended Use

Prototype component for the Smart Triage Ambulance system. Not for clinical diagnosis. Designed for real-time paramedic decision support during patient transit.

Stitch03/smart-triage-wound-cls

Smart Triage Ambulance — Wound Classification Model

Classes

Training Details

Usage

Intended Use

Stitch03/
smart-triage-wound-cls