update the entrypoint
This commit is contained in:
shuaiqing
parent
bc514523cc
commit
7168f51d42
@ -35,10 +35,14 @@ def update_exp_by_args(cfg_exp, args):
|
||||
opts_alias.append(cfg_exp.alias[args.opt_exp[i*2]])
|
||||
opts_alias.append(args.opt_exp[i*2+1])
|
||||
cfg_exp.merge_from_list(opts_alias)
|
||||
if args.skip_vis:
|
||||
if args.skip_vis or args.skip_vis_step:
|
||||
for key, val in cfg_exp.args.at_step.items():
|
||||
if key.startswith('vis'):
|
||||
val.skip = True
|
||||
if args.skip_vis or args.skip_vis_final:
|
||||
for key, val in cfg_exp.args.at_final.items():
|
||||
if key.startswith('vis'):
|
||||
val.skip = True
|
||||
|
||||
def load_cfg_from_file(cfg, args):
|
||||
cfg = Config.load(cfg)
|
||||
@ -71,6 +75,8 @@ def main_entrypoint():
|
||||
parser.add_argument('--cameras', type=str, default=None, help='Camera file path')
|
||||
parser.add_argument('--out', type=str, default=None)
|
||||
parser.add_argument('--skip_vis', action='store_true')
|
||||
parser.add_argument('--skip_vis_step', action='store_true')
|
||||
parser.add_argument('--skip_vis_final', action='store_true')
|
||||
parser.add_argument('--debug', action='store_true')
|
||||
args = parser.parse_args()
|
||||
|
||||
|
||||
200
scripts/publish/sync_Learning_EasyMocap.py
Normal file
200
scripts/publish/sync_Learning_EasyMocap.py
Normal file
@ -0,0 +1,200 @@
|
||||
filetree_l = '''
|
||||
apps:
|
||||
- mocap:
|
||||
- run.py
|
||||
myeasymocap:
|
||||
- datasets:
|
||||
- basedata.py
|
||||
- mv1p.py
|
||||
- sv1p.py
|
||||
io:
|
||||
- model.py
|
||||
- video.py
|
||||
- vis.py
|
||||
- vis3d.py
|
||||
- write.py
|
||||
operations:
|
||||
- init.py
|
||||
- loss.py
|
||||
- match.py
|
||||
- merge.py
|
||||
- optimizer.py
|
||||
- select.py
|
||||
- smooth.py
|
||||
- triangulate.py
|
||||
- match_base.py
|
||||
- iterative_triangulate.py
|
||||
stages:
|
||||
- basestage.py
|
||||
- collect.py
|
||||
backbone:
|
||||
- basetopdown.py
|
||||
- topdown_keypoints.py
|
||||
- yolo:
|
||||
- yolo.py
|
||||
vitpose:
|
||||
- layers.py
|
||||
- vit_moe.py
|
||||
hrnet:
|
||||
- __init__.py
|
||||
- modules.py
|
||||
- hrnet.py
|
||||
- myhrnet.py
|
||||
pare:
|
||||
- pare.py
|
||||
- config.py
|
||||
- constants.py
|
||||
- backbone:
|
||||
- __init__.py
|
||||
- resnet.py
|
||||
- mobilenet.py
|
||||
- hrnet.py
|
||||
- utils.py
|
||||
head:
|
||||
- __init__.py
|
||||
- pare_head.py
|
||||
- hmr_head.py
|
||||
- smpl_head.py
|
||||
- smpl_cam_head.py
|
||||
layers:
|
||||
- __init__.py
|
||||
- locallyconnected2d.py
|
||||
- interpolate.py
|
||||
- nonlocalattention.py
|
||||
- keypoint_attention.py
|
||||
- coattention.py
|
||||
- softargmax.py
|
||||
- non_local:
|
||||
- __init__.py
|
||||
- dot_product.py
|
||||
utils:
|
||||
- geometry.py
|
||||
- kp_utils.py
|
||||
mediapipe:
|
||||
- hand.py
|
||||
hand2d:
|
||||
- __init__.py
|
||||
- hand2d.py
|
||||
- resnet.py
|
||||
hmr:
|
||||
- __init__.py
|
||||
- hmr.py
|
||||
- models.py
|
||||
- hmr_api.py
|
||||
config:
|
||||
- datasets:
|
||||
- mvimage.yml
|
||||
- svimage.yml
|
||||
mv1p:
|
||||
- detect_triangulate.yml
|
||||
- detect_triangulate_fitSMPL.yml
|
||||
- detect_hand_triangulate.yml
|
||||
- detect_hand_triangulate_fitMANO.yml
|
||||
1v1p:
|
||||
- hrnet_pare_finetune.yml
|
||||
- hand_detect_finetune.yml
|
||||
- fixhand.yml
|
||||
'''
|
||||
|
||||
filetree_e = '''
|
||||
3rdparty:
|
||||
- eigen-3.3.7
|
||||
- pybind11
|
||||
library:
|
||||
- pymatch:
|
||||
- setup.py
|
||||
- CMakeLists.txt
|
||||
- python:
|
||||
- __init__.py
|
||||
- CMakeLists.txt
|
||||
- pymatchlr.cpp
|
||||
- include:
|
||||
- base.h
|
||||
- projfunc.hpp
|
||||
- Timer.hpp
|
||||
- visualize.hpp
|
||||
- matchSVT.hpp
|
||||
- pymatchlr:
|
||||
- __init__.py
|
||||
apps:
|
||||
- calibration:
|
||||
- vis_camera_by_open3d.py
|
||||
easymocap:
|
||||
- multistage:
|
||||
- gmm.py
|
||||
- gmm_08.pkl
|
||||
mytools:
|
||||
- vis_base.py
|
||||
- camera_utils.py
|
||||
'''
|
||||
|
||||
sync_config = {
|
||||
'EasyMocap': {
|
||||
'src': '../EasyMocapPublic',
|
||||
'dst': '.',
|
||||
'filetree': filetree_e
|
||||
},
|
||||
'Learn': {
|
||||
'src': '../Learning_EasyMocap',
|
||||
'dst': '.',
|
||||
'filetree': filetree_l
|
||||
}
|
||||
}
|
||||
import os
|
||||
import shutil
|
||||
import yaml
|
||||
from os.path import join
|
||||
from easymocap.mytools.debug_utils import log, mywarn
|
||||
|
||||
def compare_files(file1, file2):
|
||||
with open(file1, 'r') as f1, open(file2, 'r') as f2:
|
||||
content1 = f1.read()
|
||||
content2 = f2.read()
|
||||
if content1 == content2:
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
def copy_node(dir, nodes):
|
||||
for node in nodes:
|
||||
if isinstance(node, str):
|
||||
srcname = join(SRC, dir, node)
|
||||
dstname = join(DST, dir, node)
|
||||
if not os.path.exists(srcname):
|
||||
mywarn('Not exists {}'.format(srcname))
|
||||
continue
|
||||
if os.path.exists(dstname):
|
||||
if os.path.isdir(srcname):
|
||||
mywarn('Current not support overwrite folders: {}'.format(dstname))
|
||||
else:
|
||||
if compare_files(srcname, dstname):
|
||||
pass
|
||||
else:
|
||||
mywarn('Overwrite file: {}'.format(dstname))
|
||||
shutil.copyfile(srcname, dstname)
|
||||
else:
|
||||
if os.path.isdir(srcname):
|
||||
log('Copy dir: {}'.format(dstname))
|
||||
shutil.copytree(srcname, dstname)
|
||||
else:
|
||||
os.makedirs(join(DST, dir), exist_ok=True)
|
||||
log('Copy file: {}'.format(dstname))
|
||||
shutil.copyfile(srcname, dstname)
|
||||
elif isinstance(node, dict):
|
||||
for subdir, subnode in node.items():
|
||||
copy_node(join(dir, subdir), subnode)
|
||||
|
||||
if __name__ == '__main__':
|
||||
for cfg_name, cfg in sync_config.items():
|
||||
SRC = cfg['src']
|
||||
DST = cfg['dst']
|
||||
filetree = cfg['filetree']
|
||||
filetree = yaml.safe_load(filetree)
|
||||
SRC = os.path.abspath(SRC)
|
||||
DST = os.path.abspath(DST)
|
||||
os.chdir(SRC)
|
||||
cmd = 'git pull origin master'
|
||||
os.system(cmd)
|
||||
os.chdir(DST)
|
||||
for dir, files in filetree.items():
|
||||
copy_node(dir, files)
|
||||
Loading…
Reference in New Issue
Block a user