updates
This commit is contained in:
parent
19428ec697
commit
9315a93812
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -2,3 +2,5 @@ bazel-*
|
||||||
mediapipe/MediaPipe.xcodeproj
|
mediapipe/MediaPipe.xcodeproj
|
||||||
mediapipe/MediaPipe.tulsiproj/*.tulsiconf-user
|
mediapipe/MediaPipe.tulsiproj/*.tulsiconf-user
|
||||||
mediapipe/provisioning_profile.mobileprovision
|
mediapipe/provisioning_profile.mobileprovision
|
||||||
|
put-videos-here/*
|
||||||
|
temporary-folder
|
|
@ -0,0 +1,52 @@
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import argparse
|
||||||
|
import pathlib
|
||||||
|
# os.system('export GLOG_logtostderr=1')
|
||||||
|
# os.system('export GLOG_logtostderr=1 && bazel run --define MEDIAPIPE_DISABLE_GPU=1 \
|
||||||
|
# mediapipe/examples/desktop/hello_world:hello_world')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def main(input_data_path,output_data_path):
|
||||||
|
comp='bazel build -c opt --define MEDIAPIPE_DISABLE_GPU=1 \mediapipe/examples/desktop/hand_tracking:hand_tracking_cpu'
|
||||||
|
#명령어 컴파일
|
||||||
|
cmd='GLOG_logtostderr=1 bazel-bin/mediapipe/examples/desktop/hand_tracking/hand_tracking_cpu \--calculator_graph_config_file=mediapipe/graphs/hand_tracking/hand_tracking_desktop_live.pbtxt'
|
||||||
|
#미디어 파이프 명령어 저장listfile
|
||||||
|
listfile=os.listdir(input_data_path)
|
||||||
|
if not(os.path.isdir(output_data_path+"Relative/")):
|
||||||
|
os.mkdir(output_data_path+"Relative/")
|
||||||
|
if not(os.path.isdir(output_data_path+"Absolute/")):
|
||||||
|
os.mkdir(output_data_path+"Absolute/")
|
||||||
|
for file in listfile:
|
||||||
|
#해당 디렉토리의 하위 디렉토리 폴더명을 찾음
|
||||||
|
if not(os.path.isdir(input_data_path+file)): #ignore .DS_Store
|
||||||
|
continue
|
||||||
|
word = file+"/"
|
||||||
|
fullfilename=os.listdir(input_data_path+word)
|
||||||
|
# 하위디렉토리의 모든 비디오들의 이름을 저장
|
||||||
|
if not(os.path.isdir(output_data_path+"_"+word)):
|
||||||
|
os.mkdir(output_data_path+"_"+word)
|
||||||
|
if not(os.path.isdir(output_data_path+"Relative/"+word)):
|
||||||
|
os.mkdir(output_data_path+"Relative/"+word)
|
||||||
|
if not(os.path.isdir(output_data_path+"Absolute/"+word)):
|
||||||
|
os.mkdir(output_data_path+"Absolute/"+word)
|
||||||
|
os.system(comp)
|
||||||
|
outputfilelist=os.listdir(output_data_path+'_'+word)
|
||||||
|
for mp4list in fullfilename:
|
||||||
|
if ".DS_Store" in mp4list:
|
||||||
|
continue
|
||||||
|
inputfilen=' --input_video_path='+input_data_path+word+mp4list
|
||||||
|
outputfilen=' --output_video_path='+output_data_path+'_'+word+mp4list
|
||||||
|
cmdret=cmd+inputfilen+outputfilen
|
||||||
|
os.system(cmdret)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
cpath = pathlib.Path(__file__).parent.absolute()
|
||||||
|
# print(cpath)
|
||||||
|
input_data_path=str(cpath)+'/../put-videos-here/'
|
||||||
|
output_data_path=str(cpath)+'/../temporary-folder/'
|
||||||
|
# #print(input_data_path)
|
||||||
|
main(input_data_path,output_data_path)
|
Loading…
Reference in New Issue
Block a user