1 В избранное 0 Ответвления 0

OSCHINA-MIRROR/mirrors_trending-YOLOv6

Присоединиться к Gitlife
Откройте для себя и примите участие в публичных проектах с открытым исходным кодом с участием более 10 миллионов разработчиков. Приватные репозитории также полностью бесплатны :)
Присоединиться бесплатно
Клонировать/Скачать
Train_coco_data.md 4.4 КБ
Копировать Редактировать Web IDE Исходные данные Просмотреть построчно История
Отправлено 23.04.2025 02:07 11beda4

Обучение набора данных COCO

Эти инструкции показывают команды для обучения, чтобы воспроизвести наши результаты на наборе данных COCO.

Для мобильных моделей

YOLOv6Lite-S/M/L

python -m torch.distributed.launch --nproc_per_node 4 tools/train.py \
									--batch 128 \
									--img_size 416 \   # обучение с 416 и оценка с 320
									--conf configs/yolov6_lite/yolov6_lite_s.py \   # yolov6lite_m/l
									--data data/coco.yaml \
									--epoch 400 \
									--device 0,1,2,3 \
									--name yolov6_lite_s_coco

Для моделей P5

YOLOv6-N

# Шаг 1: Обучение базовой модели
# Убедитесь, что режим use_dfl включен в конфигурационном файле (use_dfl=True, reg_max=16)

python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 128 \
									--conf configs/yolov6n.py \
									--data data/coco.yaml \
									--epoch 300 \
									--fuse_ab \
									--device 0,1,2,3,4,5,6,7 \
									--name yolov6n_coco

# Шаг 2: Обучение с самодистилляцией
# Убедитесь, что режим use_dfl включен в конфигурационном файле (use_dfl=True, reg_max=16)
python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 128 \
									--conf configs/yolov6n.py \
									--data data/coco.yaml \
									--epoch 300 \
									--device 0,1,2,3,4,5,6,7 \
									--distill \
									--teacher_model_path runs/train/yolov6n_coco/weights/best_ckpt.pt \
									--name yolov6n_coco

YOLOv6-S/M/L

# Шаг 1: Обучение базовой модели
# Убедитесь, что режим use_dfl включен в конфигурационном файле (use_dfl=True, reg_max=16)

python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 256 \
									--conf configs/yolov6s.py \ # yolov6m/yolov6l
									--data data/coco.yaml \
									--epoch 300 \
									--fuse_ab \
									--device 0,1,2,3,4,5,6,7 \
									--name yolov6s_coco # yolov6m_coco/yolov6l_coco
```# Шаг 2: Обучение с самодистилляцией
# Убедитесь, что режим use_dfl включен в конфигурационном файле (use_dfl=True, reg_max=16)
python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 256 \ # 128 для дистилляции yolov6l
									--conf configs/yolov6s.py \ # yolov6m/yolov6l
									--data data/coco.yaml \
									--epoch 300 \
									--device 0,1,2,3,4,5,6,7 \
									--distill \
									--teacher_model_path runs/train/yolov6s_coco/weights/best_ckpt.pt \
									--name yolov6s_coco # yolov6m_coco/yolov6l_coco

Для P6 моделей

YOLOv6-N6/S6

python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 128 \
									--img 1280 \
									--conf configs/yolov6s6.py \ # yolov6n6
									--data data/coco.yaml \
									--epoch 300 \
									--bs_per_gpu 16 \
									--device 0,1,2,3,4,5,6,7 \
									--name yolov6s6_coco # yolov6n6_coco

YOLOv6-M6/L6

# Шаг 1: Обучение базовой модели
# Убедитесь, что режим use_dfl включен в конфигурационном файле (use_dfl=True, reg_max=16)
python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 128 \
									--conf configs/yolov6l6.py \ # yolov6m6
									--data data/coco.yaml \
									--epoch 300 \
									--bs_per_gpu 16 \
									--device 0,1,2,3,4,5,6,7 \
									--name yolov6l6_coco # yolov6m6_coco

# Шаг 2: Обучение с самодистилляцией
# Убедитесь, что режим use_dfl включен в конфигурационном файле (use_dfl=True, reg_max=16)
python -m torch.distributed.launch --nproc_per_node 8 tools/train.py \
									--batch 128 \
									--conf configs/yolov6l6.py \ # yolov6m6
									--data data/coco.yaml \
									--epoch 300 \
									--bs_per_gpu 16 \
									--device 0,1,2,3,4,5,6,7 \
									--distill \
									--teacher_model_path runs/train/yolov6l6_coco/weights/best_ckpt.pt \
									--name yolov6l6_coco # yolov6m6_coco

Опубликовать ( 0 )

Вы можете оставить комментарий после Вход в систему

1
https://api.gitlife.ru/oschina-mirror/mirrors_trending-YOLOv6.git
git@api.gitlife.ru:oschina-mirror/mirrors_trending-YOLOv6.git
oschina-mirror
mirrors_trending-YOLOv6
mirrors_trending-YOLOv6
main