-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathREADME.txt
More file actions
22 lines (16 loc) · 1.66 KB
/
README.txt
File metadata and controls
22 lines (16 loc) · 1.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
tools/video2feature_2.py:
Этот скрипт конвертирует исходные видеофайлы (например, .mp4, .avi, .mov) в признаки I3D,
которые сохраняются в формате .npy. Признаки I3D извлекаются с использованием предобученной модели Inception-3D (I3D),
обученной на Kinetics-400. Каждый выходной файл .npy содержит последовательность 1024-мерных векторных признаков,
представляющих всё видео.
Необходимо указать следующие флаги:
--videos_dir <путь к директории>: Путь к директории с видео.
--out_dir <путь к директории>: Путь к директории, где будут храниться i3d признаки (.npy файлы).
--model_dir <путь к директории>: Путь к библиотеке, в которой находится модель Inception-3D для извлеения i3d признаков
(на данный момент путь к библиотеке tools/pytorch-i3d)
--ckpt: Предобученная модель, находящаяся в парке models библиотеки pytorch-i3d.
Используем models/rgb_imagenet.pt, но также есть models/rgb_charades.pt, models/flow_imagenet.pt
и models/flow_charades.pt
--device: "cuda" или "cpu"
UPD:
* git clone https://github.com/piergiaj/pytorch-i3d.git tools/pytorch-i3d