Junhao Liang
bb6cd5149a
* fix bug in fix_BN * gaitgl OUMVLP support. * update ./doc/3.advance_usage.md Cross-Dataset Evalution & Data Agumentation * update config * update docs.3 * update docs.3 * add loss doc and gather input decorator * refine the create model doc * support rearrange directory of unzipped OUMVLP * fix some bugs in loss_aggregator.py * refine docs and little fix * add oumvlp pretreatment description * pretreatment dataset fix oumvlp description * add gaitgl oumvlp result * assert gaitgl input size * add pipeline * update the readme. * update pipeline and readme * Corrigendum. * add logo and remove path * update new logo * Update README.md * modify logo size Co-authored-by: 12131100 <12131100@mail.sustech.edu.cn> Co-authored-by: noahshen98 <77523610+noahshen98@users.noreply.github.com> Co-authored-by: Noah <595311942@qq.com>
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
import os
|
|
import shutil
|
|
from tqdm import tqdm
|
|
import argparse
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Test')
|
|
parser.add_argument('--input_path', default='/home1/data/OUMVLP_raw', type=str,
|
|
help='Root path of raw dataset.')
|
|
parser.add_argument('--output_path', default='/home1/data/OUMVLP_rearranged', type=str,
|
|
help='Root path for output.')
|
|
|
|
|
|
opt = parser.parse_args()
|
|
|
|
INPUT_PATH = opt.input_path
|
|
OUTPUT_PATH = opt.output_path
|
|
|
|
|
|
def mv_dir(src, dst):
|
|
shutil.copytree(src, dst)
|
|
print(src, dst)
|
|
|
|
|
|
sils_name_list = os.listdir(INPUT_PATH)
|
|
name_space = 'Silhouette_'
|
|
views = sorted(list(
|
|
set([each.replace(name_space, '').split('-')[0] for each in sils_name_list])))
|
|
seqs = sorted(list(
|
|
set([each.replace(name_space, '').split('-')[1] for each in sils_name_list])))
|
|
ids = list()
|
|
for each in sils_name_list:
|
|
ids.extend(os.listdir(os.path.join(INPUT_PATH, each)))
|
|
|
|
|
|
progress = tqdm(total=len(set(ids)))
|
|
|
|
|
|
results = list()
|
|
pid = 0
|
|
for _id in sorted(set(ids)):
|
|
progress.update(1)
|
|
for _view in views:
|
|
for _seq in seqs:
|
|
seq_info = [_id, _seq, _view]
|
|
name = name_space + _view + '-' + _seq + '/' + _id
|
|
src = os.path.join(INPUT_PATH, name)
|
|
dst = os.path.join(OUTPUT_PATH, *seq_info)
|
|
if os.path.exists(src):
|
|
try:
|
|
if os.path.exists(dst):
|
|
pass
|
|
else:
|
|
os.makedirs(dst)
|
|
for subfile in os.listdir(src):
|
|
os.symlink(os.path.join(src, subfile),
|
|
os.path.join(dst, subfile))
|
|
except OSError as err:
|
|
print(err)
|