動画ファイルに対して、認識、検出をします。
次の環境で行っています。
ubuntu 22.04
cuda 12.4
cudNN 9.1.1
Docker 26.1.3
SlowFast の動かし方
githubよりダウンロードし、ディレクトリを移動しておきます。
git clone https://github.com/eijwat/SlowFast_with_GradCam
cd SlowFast_with_GradCam
データの準備
DA_dataset
フォルダをSlowFast_with_GradCam
のディレクトリ直下に作成
mkdir DA_dataset
次の画像を参考に.mp4データを配置
(ここではフォルダ名をnega、posiで作成しています。変更する場合はmake_dataset_csv.py
内のフォルダ名も併せて変更してください)
ex
データセットの作成(配置したmp4を元にcsvを作成)
python3 make_dataset_csv.py
DA_dataset
フォルダ内にcsvファイルが作成されます
dockerのビルド
docker compose up -d
docker compose exec slowfast /bin/bash
trainの実行
python3 tools/run_net.py --cfg configs/DA_dataset/da_dataset.yaml
結果は results.csv になります。