Как мне поместить путь к папке набора данных в этот код Python? ⇐ Python

[Anonymous]

Итак, я работаю над этим проектом Python для распознавания жестов на языке жестов из git Hub.
Я проследил за прочитанным файлом и сохранил (в корневой папке проекта) все файлы набора данных в двух отдельных папках с именами train_videos и test_videos для машинного обучения.

Теперь я получаю следующую ошибку:

Код: Выделить всё

usage: video-to-frame.py [-h] gesture_folder target_folder
video-to-frame.py: error: the following arguments are required: gesture_folder, target_folder

Ниже приведен код из файла «video-to-frame.py».
Я не могу понять, куда поместить пути к моим наборам данных.

Код: Выделить всё

import cv2
import os
import pickle
from os.path import join, exists
import handsegment as hs
import argparse
from tqdm import tqdm

hc = []

def convert(gesture_folder, target_folder):
rootPath = os.getcwd()
majorData = os.path.abspath(target_folder)

if not exists(majorData):
os.makedirs(majorData)

gesture_folder = os.path.abspath(gesture_folder)

os.chdir(gesture_folder)
gestures = os.listdir(os.getcwd())

print("Source Directory containing gestures: %s" % gesture_folder)
print("Destination Directory containing frames: %s\n" % majorData)

for gesture in tqdm(gestures, unit='actions', ascii=True):
gesture_path = os.path.join(gesture_folder, gesture)
os.chdir(gesture_path)

gesture_frames_path = os.path.join(majorData, gesture)
if not os.path.exists(gesture_frames_path):
os.makedirs(gesture_frames_path)

videos = os.listdir(os.getcwd())
videos = [video for video in videos if(os.path.isfile(video))]

for video in tqdm(videos, unit='videos', ascii=True):
name = os.path.abspath(video)
cap = cv2.VideoCapture(name)  # capturing input video
frameCount = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
lastFrame = None

os.chdir(gesture_frames_path)
count = 0

# assumption only first 200 frames are important
while count < 201:
ret, frame = cap.read()  # extract frame
if ret is False:
break
framename = os.path.splitext(video)[0]
framename = framename + "_frame_" + str(count) + ".jpeg"
hc.append([join(gesture_frames_path, framename), gesture, frameCount])

if not os.path.exists(framename):
frame = hs.handsegment(frame)
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
lastFrame = frame
cv2.imwrite(framename, frame)

if cv2.waitKey(1) & 0xFF == ord('q'):
break
count += 1

# repeat last frame until we get 200 frames
while count < 201:
framename = os.path.splitext(video)[0]
framename = framename + "_frame_" + str(count) + ".jpeg"
hc.append([join(gesture_frames_path, framename), gesture, frameCount])
if not os.path.exists(framename):
cv2.imwrite(framename, lastFrame)
count += 1

os.chdir(gesture_path)
cap.release()
cv2.destroyAllWindows()

os.chdir(rootPath)

if __name__ == '__main__':
parser = argparse.ArgumentParser(description='Extract Individual Frames from gesture videos.')
parser.add_argument('gesture_folder', help='Path to folder containing folders of videos of different gestures.')
parser.add_argument('target_folder', help='Path to folder where extracted frames should be kept.')
args = parser.parse_args()
convert(args.gesture_folder, args.target_folder)

Это ссылка на репозиторий проектов на GitHub.

Подробнее здесь: https://stackoverflow.com/questions/554 ... ython-code