Это 18-й день моего участия в ноябрьском испытании обновлений. Узнайте подробности события:Вызов последнего обновления 2021 г.
окрестности
- google colab
- YOLOv5
что такое колаб
colabдаgoogleпредоставил одинjupyter notebookинструменты, поддержкаgoogle drive,tensorflowвключеныgoogleВсе семейное ведро в основном используется для разработки и исследования машинного обучения.colabНаибольшая польза для огромногоAIРазработчик предоставляет бесплатноgpuресурсы, которые можно очень легко запустить на нем, напримерtensorflow,pytorch,kerasи другие фреймворки глубокого обучения.
Обучение модели YOLOv5
приходитьgoogle drive, нажмите вверху слеваNew
Создать новую папкуcolab
Затем загрузите подготовленный набор данных маски вcolabВ папке этого набора данных мы находились вYOLOv5Модель использовалась во время обучения и может быть загружена по следующему адресу
оригинальная ссылкаpublic.Rob o flow.Love/object-of...
CSDNскачатьdownload.CSDN.net/download/Все…
Сетевой диск Baidudisk.baidu.com/is/15GSP IJ59…, Код извлечения:wja4
Далее создайтеcolab, нажмитеNew --> More --> Google Colaboratory
созданныйnotebookПосле этого вам нужно перейти в «Изменить» -> «Настройки ноутбука».gpuускорить
аппаратный ускоритель, выберитеGPU,спасти
Нажмите в правом верхнем углусоединять, выберите Подключиться к Managed Code Executor
GPUПосле настройки среды мы можемnotebookВидcolabкоторый предоставилgpuресурсы, использовать!nvidia-smiЗаказ
можно увидетьgoogleПоставляемое оборудованиеtesla P100, видеопамять 16G. Но кажется, что каждый разcolabраспределенныйgpuотличается, иногдаP100, иногдаT4, что достаточно для большинства приложений
Посмотрите нижеpytorchустановка, выполнение
import torch
torch.__version__
Вы можете видеть, что платформа была установлена по умолчанию, и версия 1.6,CUDAВерсия 10.1
Если вам нужно установить стороннюю библиотеку, вы можете установить ее прямо в ячейку, например!pip3 install torchvision
следующийgoogle driveУстановите его так, чтобы вы моглиcolabиспользуется вgoogle driveресурсы в
import os
from google.colab import drive
drive.mount('/content/drive')
path = "/content/drive/My Drive"
os.chdir(path)
os.listdir(path)
Выполните код в ячейке выше, вам будет предложено ввести код подтверждения.
google driveбудет смонтирован в каталог/content/drive, тогда ты можешьgoogle driveфайлы в
Подготовка завершена, можно скачатьYOLOv5исходный код, выполнить в ячейке
!git clone https://github.com/ultralytics/yolov5.git
затем переключитесь наgoogle drive,Исправлятьyolov5/models/yolov5s.yaml, оригиналnc: 80изменить наnc: 2
тогда ты можешь прийтиcolab,Входитьyolov5каталог для выполнения обучающих команд в ячейках
!python train.py --data ../mask/data.yaml --cfg models/yolov5s.yaml --weights '' --batch-size 64
Во время обучения,pyyamlОшибка в модуле, это связано сpyyamlПричина, по которой версия слишком низкая, может быть решена путем обновления.
pip install -U pyyaml
продолжай тренироваться
Возьми!