Openpose

👉Вернуться к списку инструкций

Openpose - первая система определения нескольких человеческих тел, рук, лиц и ступней (всего 135 ключевых точек) на отдельных изображениях в реальном времени.
Официальный сайт: https://github.com/CMU-Perceptual-Computing-Lab/openpose

Доступные версии на суперкомпьютере НИУ ВШЭ:

* 1.6 (релиз от 27.04.2020) - подключение модуля:

module load OpenPose/1.6

Тестовые исходные данные расположены в каталоге /opt/software/openpose/v1.6/examples
Пользователи: Институт когнитивных нейронаук

Выполнение расчётов на суперкомпьютере

Пример использования:

Sbatch-скрипт для запуска в очереди openpose.sbatch:

#!/bin/bash
#SBATCH --job-name=openpose             # Название задачи
#SBATCH --error=openpose-%j.err         # Файл для вывода ошибок
#SBATCH --output=openpose-%j.log        # Файл для вывода результатов
#SBATCH --time=1:00:00                  # Максимальное время выполнения
#SBATCH --ntasks 1                      # Количество MPI процессов
#SBATCH --nodes 1                       # Требуемое кол-во узлов
#SBATCH --gpus 1                        # Требуемое кол-во GPU

module load OpenPose/1.6                # Загрузка модуля OpenPose
# Выполнение расчёта на 1 ядре CPU и 1 GPU
srun openpose.bin -display 0 -image_dir /opt/software/datasets/openpose/v1.6-examples-media -model_folder $MODELS_DIR -write_images ./

Для постановки задачи в очередь выполните команду sbatch openpose.sbatch
Не забудьте скорректировать параметры в скрипте для своих расчетов! 
Посмотреть состояние своих задач можно с помощью команды mj
Сообщения, выводимые в процессе работы Openpose, будут записаны в файл openpose-00000.log , а ошибки выполнения будут записаны в файл openpose-00000.err (где 00000 - номер задачи в очереди).

Результат:

Файл COCO_val2014_000000000192_rendered.png	Файл COCO_val2014_000000000241_rendered.png