YOLOv5 🚀 на AzureML

В этом руководстве представлен быстрый старт по использованию YOLOv5 с вычислительного экземпляра AzureML.

Обрати внимание, что это руководство - краткий старт для быстрого опробования. Если ты хочешь раскрыть всю мощь AzureML, ты можешь найти документацию по нему:

• Создай актив данных
• Создайте задание AzureML
• Зарегистрируй модель

Пререквизиты

Тебе нужно рабочее пространство AzureML.

Создай вычислительный экземпляр

В рабочей области AzureML выбери Compute > Compute instances > New, выбери экземпляр с нужными тебе ресурсами.

Открой терминал

Теперь из представления "Ноутбуки" открой терминал и выбери свой комп.

Установи и запусти YOLOv5

Теперь ты можешь, создай виртуальную среду:

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

Клонируй репозиторий YOLOv5 с его подмодулями:

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

Установи необходимые зависимости:

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

Обучи модель YOLOv5 :

python train.py

Проверь модель на точность, Recall и mAP

python val.py --weights yolov5s.pt

Выполняй умозаключения на изображениях и видео:

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

Экспортируй модели в другие форматы:

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

Заметки об использовании блокнота

Обрати внимание, что если ты хочешь запускать эти команды из блокнота, то тебе нужно создать новое ядро и выбрать его в верхней части блокнота.

Если ты создашь ячейки Python , она автоматически будет использовать твое пользовательское окружение, но если ты добавишь ячейки bash , тебе нужно будет выполнить команду source activate <your-env> на каждой из этих ячеек, чтобы убедиться, что она использует твое пользовательское окружение.

Например:

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

Комментарии