Я пытаюсь использовать MMPose, чтобы найти ключевые точки в трехмерном пространстве людей в видео. Код, который я использую (который раньше работал в 2d):
Код: Выделить всё
from mmpose.apis import MMPoseInferencer
from pathlib import Path
import os
data_folder = Path("x/videos")
for filename in os.listdir(data_folder):
if not filename.endswith('.mp4'):
continue
img_path = os.path.join(data_folder, filename)
inferencer = MMPoseInferencer(pose3d="human3d")
result_generator = inferencer(img_path, out_dir='output')
Каждый раз, когда я пытаюсь получить доступ к results_generator (например, results = [result for result in result_generator]), я получаю ошибку сегмента, что, я думаю, связано с тем, что в result_generator ничего нет. Я также хотел бы, чтобы в выходной папке, которая пуста, были визуализации и данные. Есть ли что-то очевидное, что я делаю неправильно?
Подробнее здесь:
https://stackoverflow.com/questions/788 ... -mp4-files
Мобильная версия