fulyvv/EasyMocap
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:zju3dv/EasyMocap

Browse Source
This commit is contained in:
shuaiqing 2021-05-11 15:14:04 +08:00
commit a8429df920
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
requirements.txt
View File

@ -5,4 +5,5 @@ joblib
tqdm tqdm
opencv-python opencv-python
pyrender pyrender
yacs yacs
tabulate

4
scripts/preprocess/extract_video.py
View File

@ -27,7 +27,7 @@ def extract_video(videoname, path, start, end, step):
os.makedirs(outpath, exist_ok=True) os.makedirs(outpath, exist_ok=True)
video = cv2.VideoCapture(videoname) video = cv2.VideoCapture(videoname)
totalFrames = int(video.get(cv2.CAP_PROP_FRAME_COUNT)) totalFrames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
for cnt in tqdm(range(totalFrames), desc='{:-10s}'.format(os.path.basename(videoname))): for cnt in tqdm(range(totalFrames), desc='{:10s}'.format(os.path.basename(videoname))):
ret, frame = video.read() ret, frame = video.read()
if cnt < start:continue if cnt < start:continue
if cnt >= end:break if cnt >= end:break
@ -127,7 +127,7 @@ def load_openpose(opname):
def convert_from_openpose(src, dst, annotdir): def convert_from_openpose(src, dst, annotdir):
# convert the 2d pose from openpose # convert the 2d pose from openpose
inputlist = sorted(os.listdir(src)) inputlist = sorted(os.listdir(src))
for inp in tqdm(inputlist, desc='{:-10s}'.format(os.path.basename(dst))): for inp in tqdm(inputlist, desc='{:10s}'.format(os.path.basename(dst))):
annots = load_openpose(join(src, inp)) annots = load_openpose(join(src, inp))
base = inp.replace('_keypoints.json', '') base = inp.replace('_keypoints.json', '')
annotname = join(dst, base+'.json') annotname = join(dst, base+'.json')