Ultralytics YOLO27:

Dataset di pillole medicinali#

Open Medical Pills Dataset In Colab

Il dataset Medical Pills di Ultralytics è un dataset di rilevamento degli oggetti di prova di concetto (POC), composto da 115 immagini annotate appartenenti a una singola classe, pill — 92 per l'addestramento e 23 per la validazione. È stato creato per dimostrare l'uso di modelli di visione artificiale per applicazioni farmaceutiche come il controllo qualità, l'automazione del confezionamento e lo smistamento.



Watch: How to train an Ultralytics YOLO Model on the Medical Pills Detection Dataset in Google Colab

Struttura del dataset#

Il dataset Medical Pills contiene 115 immagini annotate con una singola classe, pill, suddivise in due sottoinsiemi definiti dalla configurazione medical-pills.yaml:

SuddivisioneImmaginiDescrizione
Addestramento92Immagini etichettate per l'addestramento del modello
Convalida23Immagini riservate per la valutazione e il benchmarking

Esplora Medical Pills sulla piattaforma Ultralytics per sfogliare le immagini con le relative sovrapposizioni delle annotazioni, visualizzare la distribuzione delle classi e le mappe di calore dei riquadri di delimitazione nella scheda Charts, quindi clonarlo per addestrare il tuo modello nel cloud.

Applicazioni#

L'uso della visione artificiale per il rilevamento delle pillole medicinali consente di automatizzare l'industria farmaceutica, supportando attività come:

  • Smistamento farmaceutico: automatizzare lo smistamento delle pillole in base a dimensioni, forma o colore per migliorare l'efficienza produttiva.
  • Ricerca e sviluppo nell'IA: fungere da benchmark per lo sviluppo e il test di algoritmi di visione artificiale in applicazioni farmaceutiche.
  • Sistemi digitali di inventario: alimentare soluzioni intelligenti per la gestione dell'inventario integrando il riconoscimento automatizzato delle pillole per il monitoraggio delle scorte in tempo reale e la pianificazione dei rifornimenti.
  • Controllo qualità: garantire la coerenza nella produzione delle pillole identificando difetti, irregolarità o contaminazioni.
  • Rilevamento dei contraffatti: aiutare a identificare farmaci potenzialmente contraffatti analizzandone le caratteristiche visive rispetto a standard noti.

YAML del dataset#

Il file medical-pills.yaml definisce la configurazione del dataset: i percorsi del dataset, i nomi delle classi e altri metadati. È mantenuto nel repository Ultralytics all'indirizzo https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/medical-pills.yaml.

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

Utilizzo#

Per addestrare un modello YOLO26n sul dataset Medical Pills per 100 epoche con una dimensione dell'immagine di 640, usa gli esempi seguenti. Per gli argomenti dettagliati, consulta la pagina Training del modello.

Esempio di addestramento
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)
Esempio di inferenza
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")

Immagini ed esempi di annotazioni#

Il dataset Medical Pills contiene immagini annotate che mostrano la varietà delle pillole. Di seguito è riportato un esempio di immagine annotata del dataset:

Esempio di immagine del dataset Medical Pills

  • Immagine mosaicata: viene mostrato un batch di addestramento composto da immagini mosaicata del dataset. La mosaicatura aumenta la varietà dei dati di addestramento riunendo più immagini in una sola e migliorando la capacità di generalizzazione del modello.

Integrazione con altri dataset#

Per un'analisi farmaceutica più completa, valuta la possibilità di combinare il dataset Medical Pills con altri dataset correlati, come package-seg per l'identificazione delle confezioni o dataset di imaging medico come brain-tumor, per sviluppare soluzioni di IA sanitaria end-to-end.

Citazioni e ringraziamenti#

Il dataset è disponibile ai sensi della Licenza AGPL-3.0.

Se utilizzi il dataset Medical Pills nel tuo lavoro di ricerca o sviluppo, citalo usando i dettagli indicati:

Citazione
@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#

  • Il dataset Medical Pills contiene complessivamente 115 immagini — 92 per l'addestramento e 23 per la validazione — senza una suddivisione separata per il test. Ogni immagine è annotata con una singola classe, pill. Viene fornito come download automatico di 8,19 MB definito nella configurazione medical-pills.yaml.

  • Puoi addestrare un modello YOLO26 per 100 epoche con una dimensione dell'immagine di 640px usando i metodi Python o CLI forniti. Consulta la sezione Esempio di addestramento per istruzioni dettagliate e la documentazione di YOLO26 per ulteriori informazioni sulle funzionalità del modello.

  • Il dataset consente di automatizzare il rilevamento delle pillole, contribuendo alla prevenzione della contraffazione, alla garanzia della qualità e all'ottimizzazione dei processi farmaceutici. Costituisce inoltre una risorsa preziosa per sviluppare soluzioni di IA in grado di migliorare la sicurezza dei farmaci e l'efficienza della catena di approvvigionamento.

  • L'inferenza può essere eseguita usando i metodi Python o CLI con un modello YOLO26 sottoposto a fine-tuning. Consulta la sezione Esempio di inferenza per frammenti di codice e la documentazione della modalità Predict per ulteriori opzioni.

  • Il file YAML è disponibile all'indirizzo medical-pills.yaml e contiene i percorsi del dataset, le classi e ulteriori dettagli di configurazione essenziali per addestrare modelli su questo dataset.

Commenti