Datensatz „Medical Pills“#
Der Datensatz Medical Pills von Ultralytics ist ein Machbarkeitsnachweis (POC) für die Objekterkennung mit 115 annotierten Bildern und einer einzigen Klasse, pill – 92 für das Training und 23 für die Validierung. Er wurde entwickelt, um Computer-Vision-Modelle für pharmazeutische Anwendungen wie Qualitätskontrolle, automatisierte Verpackung und Sortierung zu demonstrieren.
Watch: How to train an Ultralytics YOLO Model on the Medical Pills Detection Dataset in Google Colab
Datensatzstruktur#
Der Datensatz Medical Pills enthält 115 Bilder, die mit einer einzigen Klasse annotiert sind: pill. Sie sind gemäß der Konfiguration medical-pills.yaml in zwei Teilmengen aufgeteilt:
| Aufteilung | Bilder | Beschreibung |
|---|---|---|
| Trainieren | 92 | Beschriftete Bilder für das Modelltraining |
| Validierung | 23 | Zurückgehaltene Bilder für die Evaluierung und das Benchmarking |
Erkunde Medical Pills auf der Ultralytics Platform, um die Bilder mit ihren Annotationen zu durchsuchen, die Klassenverteilung und Heatmaps der Begrenzungsrahmen im Tab Charts anzuzeigen und den Datensatz zu klonen, um in der Cloud dein eigenes Modell zu trainieren.
Anwendungen#
Der Einsatz von Computer Vision zur Erkennung medizinischer Tabletten ermöglicht die Automatisierung in der pharmazeutischen Industrie und unterstützt unter anderem folgende Aufgaben:
- Pharmazeutische Sortierung: Automatisierte Sortierung von Tabletten nach Größe, Form oder Farbe zur Steigerung der Produktionseffizienz.
- Forschung und Entwicklung im Bereich KI: Der Datensatz dient als Benchmark für die Entwicklung und Erprobung von Computer-Vision-Algorithmen in pharmazeutischen Anwendungsfällen.
- Digitale Bestandsverwaltung: Intelligente Lösungen für die Bestandsverwaltung durch die Integration automatisierter Tablettenerkennung zur Bestandsüberwachung in Echtzeit und zur Planung der Nachbestellung.
- Qualitätskontrolle: Sicherstellung einer gleichbleibenden Qualität bei der Tablettenproduktion durch die Erkennung von Defekten, Unregelmäßigkeiten oder Verunreinigungen.
- Erkennung von Fälschungen: Unterstützung bei der Identifizierung potenziell gefälschter Medikamente durch den Abgleich visueller Merkmale mit bekannten Standards.
Dataset-YAML#
Die Datei medical-pills.yaml definiert die Datensatzkonfiguration – die Pfade zum Datensatz, die Klassennamen und weitere Metadaten. Sie wird im Ultralytics-Repository unter https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/medical-pills.yaml gepflegt.
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# Medical-pills dataset by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/detect/medical-pills
# Example usage: yolo train data=medical-pills.yaml
# parent
# ├── ultralytics
# └── datasets
# └── medical-pills ← downloads here (8.19 MB)
# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: medical-pills # dataset root dir
train: images/train # train images (relative to 'path') 92 images
val: images/val # val images (relative to 'path') 23 images
# Classes
names:
0: pill
# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/medical-pills.zipVerwendung#
Um ein YOLO26n-Modell 100 Epochen lang mit einer Bildgröße von 640 auf dem Datensatz Medical Pills zu trainieren, verwende die folgenden Beispiele. Ausführliche Argumente findest du auf der Trainingsseite des Modells.
from ultralytics import YOLO
# Load a model
model = YOLO("yolo26n.pt") # load a pretrained model (recommended for training)
# Train the model
results = model.train(data="medical-pills.yaml", epochs=100, imgsz=640)from ultralytics import YOLO
# Load a model
model = YOLO("path/to/best.pt") # load a fine-tuned model
# Inference using the model
results = model.predict("https://ultralytics.com/assets/medical-pills-sample.jpg")Beispielbilder und Annotationen#
Der Datensatz Medical Pills enthält annotierte Bilder, die die Vielfalt der Tabletten zeigen. Unten siehst du ein Beispiel für ein annotiertes Bild aus dem Datensatz:

- Mosaikbild: Zu sehen ist ein Trainingsstapel aus zu einem Mosaik zusammengesetzten Datensatzbildern. Die Mosaikdarstellung erhöht die Vielfalt der Trainingsdaten, indem mehrere Bilder zu einem Bild zusammengeführt werden, und verbessert dadurch die Generalisierungsfähigkeit des Modells.
Integration mit anderen Datensätzen#
Für eine umfassendere pharmazeutische Analyse kannst du den Datensatz Medical Pills mit anderen verwandten Datensätzen wie package-seg zur Identifizierung von Verpackungen oder mit Datensätzen aus der medizinischen Bildgebung wie brain-tumor kombinieren, um durchgängige KI-Lösungen für das Gesundheitswesen zu entwickeln.
Zitate und Danksagungen#
Der Datensatz ist unter der AGPL-3.0-Lizenz verfügbar.
Wenn du den Datensatz Medical Pills in deiner Forschungs- oder Entwicklungsarbeit verwendest, zitiere ihn bitte anhand der genannten Angaben:
@dataset{Jocher_Ultralytics_Datasets_2024,
author = {Jocher, Glenn and Rizwan, Muhammad},
license = {AGPL-3.0},
month = {Dec},
title = {Ultralytics Datasets: Medical-pills Detection Dataset},
url = {https://docs.ultralytics.com/datasets/detect/medical-pills},
version = {1.0.0},
year = {2024}
}FAQ#
Der Datensatz Medical Pills enthält insgesamt 115 Bilder – 92 für das Training und 23 für die Validierung – ohne separate Testaufteilung. Jedes Bild ist mit einer einzigen Klasse annotiert:
pill. Er wird als automatischer Download mit einer Größe von 8,19 MB bereitgestellt, der in der Konfigurationmedical-pills.yamldefiniert ist.Du kannst ein YOLO26-Modell 100 Epochen lang mit einer Bildgröße von 640 px anhand der bereitgestellten Python- oder CLI-Methoden trainieren. Ausführliche Anweisungen findest du im Abschnitt Trainingsbeispiel. Weitere Informationen zu den Modellfunktionen enthält die YOLO26-Dokumentation.
Der Datensatz ermöglicht die Automatisierung der Tablettenerkennung und trägt zur Vermeidung von Fälschungen, zur Qualitätssicherung und zur Optimierung pharmazeutischer Prozesse bei. Außerdem ist er eine wertvolle Ressource für die Entwicklung von KI-Lösungen, die die Arzneimittelsicherheit und die Effizienz der Lieferkette verbessern können.
Die Inferenz kann mit Python- oder CLI-Methoden und einem fein abgestimmten YOLO26-Modell durchgeführt werden. Codebeispiele findest du im Abschnitt Inferenzbeispiel; weitere Optionen beschreibt die Dokumentation zum Predict-Modus.
Die YAML-Datei ist unter medical-pills.yaml verfügbar. Sie enthält die Pfade zum Datensatz, die Klassen und weitere für das Training von Modellen auf diesem Datensatz erforderliche Konfigurationsdetails.



