1
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have passed since last update.

【初心者】YOLOv5を使ってみる #1 (AWS上でのインストール、物体検出確認)

Posted at

1. 目的

  • 画像/映像分析の勉強のため、簡単に物体検出ができるフレームワークであるYOLOv5を使ってみて基本を理解する。

2. やったこと

  • AWS上のWindows Server 2019にYOLOv5をインストールする。
  • あらかじめ保存した画像ファイル、動画ファイルに対して推論を行い、結果を出力できることを確認する。

3. 構成

  • 単純にWindows Server 2019のインスタンスを立てる。今回は学習の処理は行わないため、GPUインスタンスではなく普通のCPUのみのインスタンスを使用。

構成図1.png

4. 実施手順

4.1 事前準備

  • Windows Server 2019の起動

    • 普通にWindows Server 2019を起動する。(RDPして、OSの中でWindowを開いて動画再生したりしたかったため、今回はOSとしてWindowsを選択)
  • Gitのインストール

    • YOLOのインストールに必要なため、Git for Windowsをインストールする。
    • ダウンロードサイト: https://gitforwindows.org/
    • 今回のバージョン: Git-2.33.0-64-bit
  • Python のインストール

    • YOLOの動作に必要なため、Pythonをインストールする。
    • ダウンロードサイト: https://www.python.org/downloads/
    • 今回のバージョン: python 3.9.6
    • 注意点: 基本的にデフォルトのままインストールするが、「Add Python 3.9 to PATH」(パスを通す設定)だけチェックを入れる。(下図)

yolo01a.png

4.2 YOLOv5のインストール

  • YOLOv5のインストール
    • 公式サイトの「Getting Started」の手順通りにインストールする。
C:\Users\Administrator>git clone https://github.com/ultralytics/yolov5.git
C:\Users\Administrator>cd yolov5
C:\Users\Administrator\yolov5>pip install -r requirements.txt

4.3 動作確認

画像の物体検出

  • サンプルとして用意されている画像を指定して物体検出を行う。
C:\Users\Administrator\yolov5>python detect.py --source "C:\Users\Administrator\yolov5\data\images\zidane.jpg"
  • C:\Users\Administrator\yolov5\runs\detect\expX (expXのサブフォルダは試行毎に作成される)に結果が保存される。

yolo02a.png

映像の物体検出

C:\Users\Administrator\yolov5>python detect.py --source "C:\Users\Administrator\yolov5\winter.mp4"
  • 映像についても画像と同様に物体検出が可能。

yolo03.png

5. 所感

  • 超今更ながらYOLO入門できたので、モデルの作成など引き続き深堀していきたい。
1
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
1
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?