Ultralytics YOLO27:
Get Started

Datensatz für medizinische Pillen#

Open Medical Pills Dataset In Colab

Der Ultralytics-Datensatz Medical Pills ist ein Machbarkeitsnachweis (POC) für die Objekterkennung. Er umfasst 115 gelabelte Bilder mit einer einzigen Klasse, pill — 92 für das Training und 23 für die Validierung. Der Datensatz veranschaulicht den Einsatz von Computer-Vision-Modellen in pharmazeutischen Anwendungen wie Qualitätskontrolle, Verpackungsautomatisierung und Sortierung.



Ansehen: So trainierst du ein Ultralytics-YOLO-Modell mit dem Datensatz zur Erkennung medizinischer Tabletten in Google Colab

Datensatzstruktur#

Der Medical-Pills-Datensatz enthält 115 Bilder, die einer einzigen Klasse, pill, zugeordnet und in zwei durch die Konfiguration medical-pills.yaml definierte Teilmengen aufgeteilt sind:

AufteilungBilderBeschreibung
Trainieren92Annotierte Bilder für das Modelltraining
Validierung23Zurückgehaltene Bilder für die Evaluierung und das Benchmarking

Erkunde Medical Pills auf der Ultralytics Platform, um die Bilder mit eingeblendeten Annotationen zu durchsuchen, die Klassenverteilung und Begrenzungsrahmen-Wärmekarten im Tab Charts anzusehen und den Datensatz zu klonen, um dein eigenes Modell in der Cloud zu trainieren.

Anwendungsfälle#

Der Einsatz von Computer Vision zur Erkennung medizinischer Pillen ermöglicht die Automatisierung in der Pharmaindustrie und unterstützt Aufgaben wie:

  • Sortierung von Arzneimitteln: Automatisiere die Sortierung von Pillen nach Größe, Form oder Farbe, um die Produktionseffizienz zu steigern.
  • KI-Forschung und -Entwicklung: Dient als Benchmark für die Entwicklung und Erprobung von Computer-Vision-Algorithmen für pharmazeutische Anwendungsfälle.
  • Digitale Bestandssysteme: Ermöglicht intelligente Lösungen zur Bestandsverwaltung durch automatisierte Pillenerkennung für die Echtzeitüberwachung des Lagerbestands und die Planung der Nachbestellung.
  • Qualitätskontrolle: Sichert die gleichbleibende Qualität der Pillenproduktion, indem Fehler, Unregelmäßigkeiten oder Verunreinigungen erkannt werden.
  • Erkennung gefälschter Arzneimittel: Hilft dabei, potenziell gefälschte Medikamente zu erkennen, indem ihre visuellen Merkmale mit bekannten Standards verglichen werden.

YAML-Datei des Datensatzes#

Die Datei medical-pills.yaml definiert die Dataset-Konfiguration — die Dataset-Pfade, 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/cfg/datasets/medical-pills.yaml
# 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.zip

Verwendung#

Verwende die folgenden Beispiele, um ein YOLO26n-Modell 100 Epochen lang mit einer Bildgröße von 640 auf dem Medical-Pills-Datensatz zu trainieren. Ausführliche Angaben zu den Argumenten findest du auf der Seite Training des Modells.

Trainingsbeispiel
from ultralytics import YOLO

# Ein Modell laden
model = YOLO("yolo26n.pt")  # ein vortrainiertes Modell laden (für das Training empfohlen)

# Das Modell trainieren
results = model.train(data="medical-pills.yaml", epochs=100, imgsz=640)
Inferenzbeispiel
from ultralytics import YOLO

# Ein Modell laden
model = YOLO("path/to/best.pt")  # Ein feinabgestimmtes Modell laden

# Inferenz mit dem Modell
results = model.predict("https://ultralytics.com/assets/medical-pills-sample.jpg")

Beispielbilder und Annotationen#

Der Medical-Pills-Datensatz enthält gelabelte Bilder, die die Vielfalt der Pillen zeigen. Unten siehst du ein Beispiel für einen gekachelten Trainingsbatch aus dem Datensatz:

Beispielbild aus dem Medical-Pills-Datensatz

  • Mosaikbild: Zu sehen ist ein Trainingsbatch aus mosaikierten Dataset-Bildern. Die Mosaikdarstellung erhöht die Vielfalt der Trainingsdaten, indem mehrere Bilder zu einem zusammengefügt werden, wodurch die Generalisierungsfähigkeit des Modells verbessert wird.

Integration mit anderen Datensätzen#

Für umfassendere pharmazeutische Analysen kannst du den Datensatz Medical Pills mit anderen verwandten Datensätzen kombinieren, etwa mit package-seg zur Identifizierung von Verpackungen oder mit Bildgebungsdatensätzen aus der Medizin wie brain-tumor, 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 für deine Forschung oder Entwicklungsarbeit verwendest, zitiere ihn bitte mit den genannten Angaben:

Zitat
@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}
}

Häufig gestellte Fragen#

  • Der Datensatz Medical Pills umfasst insgesamt 115 Bilder – 92 für das Training und 23 für die Validierung – und hat keinen separaten Testsatz. Jedes Bild ist mit einer einzigen Klasse annotiert: pill. Der Datensatz wird als automatischer Download mit einer Größe von 8,19 MB bereitgestellt, der in der Konfiguration medical-pills.yaml definiert ist.

  • Du kannst ein YOLO26-Modell mit den bereitgestellten Python- oder CLI-Methoden 100 Epochen lang mit einer Bildgröße von 640 px 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 Pillenerkennung und trägt so zur Fälschungsprävention, Qualitätssicherung und 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 lässt sich mit Python- oder CLI-Methoden und einem feinabgestimmten YOLO26-Modell durchführen. Codebeispiele findest du im Abschnitt Inferenzbeispiel; weitere Optionen werden in der Dokumentation zum Vorhersagemodus beschrieben.

  • Die YAML-Datei ist unter medical-pills.yaml verfügbar. Sie enthält Dataset-Pfade, Klassen und weitere Konfigurationsdetails, die für das Training von Modellen mit diesem Datensatz erforderlich sind.

Kommentare