Overslaan naar inhoud

YOLOv5 🚀 op AzureML

Deze handleiding geeft een quickstart voor het gebruik van YOLOv5 vanuit een AzureML compute instance.

Merk op dat deze handleiding een quickstart is om snel uit te proberen. Als je de volledige kracht van AzureML wilt ontsluiten, kun je de documentatie vinden op:

Vereisten

Je hebt een AzureML workspace nodig.

Een computerinstantie maken

Selecteer in je AzureML werkruimte Compute > Compute instances > New en selecteer de instance met de resources die je nodig hebt.

maak-computiepijl

Terminal openen

Open nu vanuit de Notebookweergave een Terminal en selecteer je computer.

open-terminal-pijl

Instellen en uitvoeren YOLOv5

Nu kun je een virtuele omgeving creëren:

conda create --name yolov5env -y
conda activate yolov5env
conda install pip -y

Kloon YOLOv5 repository met zijn submodules:

git clone https://github.com/ultralytics/yolov5
cd yolov5
git submodule update --init --recursive # Note that you might have a message asking you to add your folder as a safe.directory just copy the recommended command

Installeer de vereiste afhankelijkheden:

pip install -r yolov5/requirements.txt
pip install onnx>=1.10.0

Train het YOLOv5 model:

python train.py

Valideer het model voor Precisie, Recall en mAP

python val.py --weights yolov5s.pt

Inferentie uitvoeren op afbeeldingen en video's:

python detect.py --weights yolov5s.pt --source path/to/images

Modellen exporteren naar andere formaten:

python detect.py --weights yolov5s.pt --source path/to/images

Opmerkingen over het gebruik van een notitieboek

Merk op dat als je deze commando's vanuit een notitieboek wilt uitvoeren, je een nieuwe Kernel moet maken en je nieuwe Kernel bovenaan je notitieboek moet selecteren.

Als je Python cellen maakt, wordt automatisch je aangepaste omgeving gebruikt, maar als je bash cellen toevoegt, moet je het volgende uitvoeren source activate <your-env> op elk van deze cellen om er zeker van te zijn dat het jouw aangepaste omgeving gebruikt.

Bijvoorbeeld:

%%bash
source activate newenv
python val.py --weights yolov5s.pt


Aangemaakt 2023-11-12, Bijgewerkt 2024-06-02
Auteurs: glenn-jocher (3), ouphi (1)

Reacties