Arguments parser in hpe_json.py

This commit is contained in:
Arkadiy Strelnikov
2021-11-29 14:28:49 +07:00
parent 5459e0f0ce
commit bee2dfa944
3 changed files with 540 additions and 20 deletions

View File

@@ -2,28 +2,28 @@
## Запуск
Для вебкамеры, где 0 - индекс самой вебкамеры, если отличается, то следует заменить на нужный.
Для тестового запуска через вебкамеру с индексом 0.
```console
python3 hpe_videocapture.py 0 scale_pose
python3 hpe_json.py --address_input 0 --show_image True
```
Для чтения из файла
Скрипт принимает следующие аргументы:
* '--address_input' (type=str, default="0"): путь к файлу или индекс вебкамеры
* '--scale_pose' (type=float, default=0.42): ширина плеч в метрической системе
* '--crop_image' (type=float, default=1.0): коэффициент, если нужно изменить размер изображения
* '--osc_address' (type=str, default="0.0.0.0"): ip-адрес для osc клиента
* '--osc_port' (type=int, default="5005"): порт для osc клиента
* '--osc_message_address' (type=str, default="/pose/0"): адрес для вывода сообщения через osc клиент
* '--output_method' (type=str, default='file', choices=['file', 'osc']): тип вывода через файл или osc клиент
* '--mirror_image' (type=bool, default=True): горизнтальное отображение выходного изображения
* '--show_image' (type=bool, default=False): вывод выходного изображения через opencv
Для теста с matplotlib нужно запустить скрипт (пример с вебкамерой с индексом 0):
```console
python3 hpe_videocapture.py path_to_video scale_pose
python3 hpe_videocapture.py 0
```
Параметр scale_pose - ширина плеч. По умолчанию равен 0,42.
Есть экспериментальная версия для взаимодействия с ffmpeg и osc:
```console
python3 hpe_ffmpeg.py
```
Внутри кода записаны параметры для osc соединения:
* osc_address = "10.199.0.14"
* osc_port = 5005
Предполагается вывести выше указанные параметры либо в аргументы, либо в конфиг-файл.
## Зависимости
Библиотеки python:
@@ -32,7 +32,7 @@ python3 hpe_ffmpeg.py
* matplotlib (можно сделать опциональной)
* opencv-python
* json
* python-osc (в файле hpe_ffmpeg.py)
* python-osc
Библиотека mediapipe требует cudatoolkit и cudnn для работы с gpu.
@@ -76,6 +76,3 @@ python3 hpe_ffmpeg.py
* hol_smooth_landmarks = True
* hol_min_detection_confidence = 0.7
* hol_min_tracking_confidence = 0.5
Физические параметры человека (ширина плеч в метрах):
* scale_pose = 0,42