87 lines
2.9 KiB
YAML
87 lines
2.9 KiB
YAML
module: myeasymocap.stages.basestage.MultiStage
|
||
args:
|
||
output: output/detect_match_triangulate
|
||
keys_keep: [cameras, imgnames]
|
||
at_step:
|
||
detect:
|
||
module: myeasymocap.backbone.yolo.yolo.MultiPerson # Use YOLO to detect multi-person
|
||
key_from_data: [images, imgnames]
|
||
args:
|
||
model: yolov5m
|
||
name: person
|
||
min_length: 150 # this two threshold control the wanted bboxes
|
||
max_length: 1000
|
||
# keypoints2d:
|
||
# module: myeasymocap.backbone.hrnet.myhrnet.MyHRNet
|
||
# key_from_data: [images, imgnames]
|
||
# key_from_previous: [bbox]
|
||
# key_keep: []
|
||
# args:
|
||
# ckpt: data/models/pose_hrnet_w48_384x288.pth
|
||
# single_person: False # This flag controls the function to detect all keypoints
|
||
keypoints2d:
|
||
module: myeasymocap.backbone.vitpose.vit_moe.MyViT
|
||
key_from_data: [images, imgnames]
|
||
key_from_previous: [bbox]
|
||
key_keep: []
|
||
args:
|
||
ckpt: data/models/vitpose+_base.pth
|
||
single_person: False # This flag controls the function to detect all keypoints
|
||
vis_2d:
|
||
module: myeasymocap.io.vis.Vis2D
|
||
skip: False
|
||
key_from_data: [images]
|
||
key_from_previous: [keypoints, bbox]
|
||
args:
|
||
name: vis_keypoints2d
|
||
scale: 0.5
|
||
match:
|
||
module: myeasymocap.operations.match_base.MatchAndTrack
|
||
key_from_data: [cameras, meta]
|
||
key_from_previous: [keypoints]
|
||
args:
|
||
cfg_match:
|
||
min_conf: 0.3
|
||
min_joints: 9
|
||
distance:
|
||
mode: epipolar
|
||
threshold: 0.05 # 用于控制匹配的内点阈值
|
||
threshold_track: 0.05 # track的匹配的内点阈值
|
||
min_common_joints: 9
|
||
cfg_svt:
|
||
debug: 0
|
||
maxIter: 10
|
||
w_sparse: 0.1
|
||
w_rank: 50
|
||
tol: 0.0001
|
||
aff_min: 0.3
|
||
triangulate:
|
||
min_view: 3 # min view when triangulate each points
|
||
min_view_body: 3 # min visible view of the body
|
||
min_conf_3d: 0.1
|
||
dist_max: 50 # pixel
|
||
dist_track: 100 # mm
|
||
cfg_track:
|
||
max_person: 100
|
||
max_missing: 3 # 最多丢失3帧就要删除
|
||
final_ranges: [[-10000, -10000, -10000], [10000, 10000, 10000]] # 最终的输出的range，仅用于输出的时候的筛选
|
||
final_max_person: 100
|
||
kintree: [[2, 3], [5, 6], [3, 4], [6, 7], [11, 22], [22, 23], [11, 24], [14, 19], [19, 20], [14, 21]]
|
||
vis_kpts3d:
|
||
module: myeasymocap.io.vis.Vis3D
|
||
key_from_data: [images, cameras]
|
||
key_from_previous: [results] # 用于最后的一起优化
|
||
args:
|
||
scale: 0.5
|
||
lw_factor: 10
|
||
at_final:
|
||
write_raw: # write the raw 3d keypoints
|
||
module: myeasymocap.io.write.WriteAll
|
||
key_from_data: [results, meta]
|
||
args:
|
||
name: keypoints3d_raw
|
||
make_video:
|
||
module: myeasymocap.io.video.MakeVideo
|
||
args:
|
||
fps: 60
|
||
keep_image: False |