金子邦彦研究室人工知能Windows で動く人工知能関係 Pythonアプリケーション,オープンソースソフトウエア)AlexryAB/darknet のインストールと動作確認(Scaled YOLO v4 による物体検出)(Windows 上)

AlexryAB/darknet のインストールと動作確認(Scaled YOLO v4 による物体検出)(Windows 上)

AlexeyAB darknet のインストールと動作確認(Scaled YOLO v4 による物体検出)を行う. 重みのデータ yolov4-p6.weights を使用

目次

  1. 前準備
  2. github の AlexeyAB/darknet のインストール
  3. darknet を用いた物体検出

AlexeyAB darknet

AlexeyAB darknet は,YOLOv2, YOLOv3, YOLOv4 の機能などを持つ.

文献

Chien-Yao Wang, Alexey Bochkovskiy, Hong-Yuan Mark Liao, Scaled-YOLOv4: Scaling Cross Stage Partial Network, Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) 2021, pp. 13029-13038, also CoRR, Scaled-YOLOv4: Scaling Cross Stage Partial Network, 2021.

https://arxiv.org/pdf/2011.08036v2.pdf

関連する外部ページ

関連項目YOLOv3, YOLOv4, RetinaNet, 物体検出

前準備

Git のインストール(Windows 上)

Gitは,バージョン管理システム.ソースコードの管理や複数人での共同に役立つ.

サイト内の関連ページ

Windows での Git のインストール: 別ページ »で説明している.

関連する外部ページ

Git の公式ページ: https://git-scm.com/

CMake のインストール(Windows 上)

CMake はビルドツールである.

サイト内の関連ページ

Windows での cmake のインストール: 別ページ »で説明している.

関連する外部ページ

CMake の公式ダウンロードページ: https://cmake.org/download/

Build Tools for Visual Studio 2022,NVIDIA ドライバ,NVIDIA CUDA ツールキット 11.8,NVIDIA cuDNN 8.6 のインストール(Windows 上)

サイト内の関連ページ

NVIDIA グラフィックスボードを搭載しているパソコンの場合には, NVIDIA ドライバNVIDIA CUDA ツールキットNVIDIA cuDNN のインストールを行う.

関連する外部ページ

github の AlexeyAB/darknet のインストール

OpenCV のインストール(Windows 上)

OpenCV 4.7.0 のインストール,動作確認(Windows 上): 別ページ »で説明している.

github の AlexeyAB/darknet のインストール

URL: https://github.com/AlexeyAB/darknet

Windows での手順を下に示す.Ubuntu でも同様の手順になる.

  1. Windows で,コマンドプロンプトを実行.
  2. インストールディレクトリを空にする
    mkdir %HOMEPATH%
    cd %HOMEPATH%
    rmdir /s /q darknet
    

    [image]
  3. github の AlexeyAB/darknet のダウンロード

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

    [image]
  4. darknet のインストール

    https://github.com/AlexeyAB/darknet に記載の手順による

    しばらく待つ.

    cd %HOMEPATH%
    cd darknet
    powershell -Command "Set-ExecutionPolicy unrestricted -Scope CurrentUser -Force"
    powershell -Command ".\build.ps1 -UseVCPKG -EnableOPENCV -EnableCUDA -EnableCUDNN"
    

    [image]
    (以下省略)
  5. 結果の確認

    エラーメッセージが出ていないことを確認

    [image]

darknet を用いた物体検出

darknet のソースコードなどの URL: https://github.com/AlexeyAB/darknet

COCO データセットで事前学習済みモデルの重みのデータの URL: https://github.com/AlexeyAB/darknet

重みのデータ yolov4-p6.weights を使用

  1. 重みのデータ yolov4-p6.weights のダウンロード

    https://github.com/AlexeyAB/darknet の記載に従う.

    [image]
  2. パスを通す

    Windows では,システム環境変数 PATH に,次の2つを追加.

  3. 動画ファイルを準備する

    ここでは,動画ファイル https://www.kkaneko.jp/sample/video/samplevideo.mp4 を使用することにする.

    Windows で,コマンドプロンプトを実行.

    cd %HOMEPATH%
    cd darknet
    curl -O https://www.kkaneko.jp/sample/video/samplevideo.mp4
    

    [image]
  4. 実行
    cd %HOMEPATH%\darknet
    darknet detector demo cfg\coco.data cfg\yolov4-p6.cfg yolov4-p6.weights samplevideo.mp4
    

    [image]
  5. 表示を確認

    [image]

    [image]