Обнаружение объектов YOLOv4

глубокое обучение

Это 28-й день моего участия в ноябрьском испытании обновлений. Узнайте подробности события:Вызов последнего обновления 2021 г.

Программно-аппаратная среда

  • ubuntu 18.04 64bit
  • nvidia gtx 1660
  • YOLOv4

YOLOv4

YOLOЭто один из отличных алгоритмов в области обнаружения целей.YOLOВыпущена четвертая версия, адрес бумаги:АР Вест V.org/PDF/2004.10…, статья предлагает пять основных улучшений, более двадцати техник, эффект замечательный, такой же,YOLOv4также на основеdarknetРамка.

Обнаружение YOLOv4

Исходный код проекта размещен по адресуGitHub.com/Alex Malicious AB/Большой…, за основным использованием следуетYOLOv3Это точно так же, если вы не уверены, вы можете обратиться к предыдущей статье.Основное использование даркнета

Сначала нам нужно скачатьYOLOv4файл модели вdrive.Google.com/open?ID=1Проверить…, если ссылка не открывается, перейдите по ссылке ниже, чтобы скачать

Далее скачиваем исходный код

git clone https://github.com/AlexeyAB/darknet.git
cd darknet

по умолчанию,MakefileНе открытоgpu,cudnnиopencvПодожди вариантов, у меня эти окружения на боку, примерноcuda,cudnnДля установки вы можете обратиться к предыдущей ссылкеImagination.com/2019/12/13/…

Таким образом, вам все равно придется редактировать его в соответствии с вашей реальной ситуацией.Makefile,Исправлять

GPU=1
CUDNN=1
OPENCV=1

После успешной модификации начните компиляцию и введите команду напрямуюmake

make

Дождитесь окончания процесса компиляции, исполняемый файл будет сгенерирован в текущем каталогеdarknet

Наконец, давайте протестируем

# 图片检测
./darknet detect cfg/yolov4.cfg yolov4.weights data/dog.jpg

yolov4

# 图片检测
./darknet detect cfg/yolov4.cfg yolov4.weights data/horses.jpg

yolov4

КонтрастYOLOv3Результат обнаружения можно найти интуитивно,YOLOv4в состоянии обнаружитьdog.jpgв правом верхнем углуpottedplant,иYOLOv3нет; вhorses.jpgсередина,YOLOv4может обнаружить всех лошадей, иYOLOv3Только 4 могут быть обнаружены

# 本地视频文件测试
./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights ~/Videos/test.mp4

# 网络摄像头实时测试
./darknet detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights rtsp://admin:admin12345@192.168.0.100:554 -i 0 -thresh 0.25

использованная литература

Категории