Skip to content

albivaltzew/DsworksEqualAI

Repository files navigation

DsworksEqualAI

DSworks AIJ 2023 Equal AI Competition

  1. Датасет содержит 1000 классов жестов РЖЯ по 20 видео в каждом классе = 20000 видео
  2. В разметке есть начало и конец жеста - обрежем видео для удаления ненужных кадров и уменьшения размера датасета
  3. Переводим текст в метку класса
  4. Делим датасет на train, val, с условием, что человек из train не принадлежит val
  5. Дообучаем baseline модель
  6. Есть вариант обучить легковесную модель TVN (https://github.com/DELTA37/TVN)

Результат: Классификация видео является ресурсозатратной задачей и обучить нейросеть from scratch, как сказано в условии, в отсутствии железа не представляется реализуемым. В подобных задачах является нормальной практикой файнтюнить модель на своих данных.

Install MMAction2 locally with cuda in conda environment:

Install cuda win10 https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

  1. conda create --name openmmlab python=3.8 -y conda activate openmmlab
  2. conda install cuda -c nvidia/label/cuda-12.1.0
  3. nvcc -V
  4. conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
  5. python -m torch.utils.collect_env
  6. import torch; print(torch.version.cuda); print(torch.cuda.is_available()); print(torch.backends.cudnn.enabled)
  7. conda install -c anaconda git
  8. git –-version
  9. pip install -U openmim
  10. mim install mmengine
  11. mim install mmcv
  12. git clone https://github.com/open-mmlab/mmaction2.git

cd mmaction2

pip install -v -e .

  1. mim download mmaction2 --config tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb --dest .

  2. python demo/demo.py tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb.py tsn_imagenet-pretrained-r50_8xb32-1x1x8-100e_kinetics400-rgb_20220906-2692d16c.pth demo/demo.mp4 tools/data/kinetics/label_map_k400.txt

About

Dsworks AIJ 2023 Equal AI Competition

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published