MultiScale3D/ensemble.py at main · Zhai-Mao/MultiScale3D · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
from mmcv import load, dump
from tools.smp import *
import datetime

joint_path = '/root/autodl-tmp/pyskl-main/data/Weizmann/test-res.json'
bone_path = '/root/autodl-tmp/pyskl-main/data/Weizmann/gym_limb.json'
# kbone_path = '../work_dirs/ntu60_xsub/k/best_pred.pkl'
# joint_motion_path = '../work_dirs/ntu60_xsub/jm/best_pred.pkl'
# bone_motion_path = '../work_dirs/ntu60_xsub/bm/best_pred.pkl'
# kbone_motion_path = '../work_dirs/ntu60_xsub/km/best_pred.pkl'

joint = load(joint_path)
bone = load(bone_path)
# kbone = load(kbone_path)
# joint_motion = load(joint_motion_path)
# bone_motion = load(bone_motion_path)
# kbone_motion = load(kbone_motion_path)

label = load_label('/root/autodl-tmp/pyskl-main/tools/data/gym_hrnet.pkl', 'val')
# label = load_label('/data/nturgbd/ntu60_3danno.pkl', 'xview_val')
# label = load_label('/data/nturgbd/ntu120_3danno.pkl', 'xsub_val')
# label = load_label('/data/nturgbd/ntu120_3danno.pkl', 'xset_val')
# label = load_label('/data/k400/k400_hrnet.pkl', 'val')
# label = load_label('/data/finegym/gym_hrnet.pkl', 'val')

print('J+B')

fused = comb([joint, bone], [1, 1])
print('Top-1', top1(fused, label))

# print('4M')
# fused = comb([joint, bone, joint_motion, bone_motion], [2, 2, 1, 1])
# print('Top-1', top1(fused, label))

# print('6M')
# fused = comb([joint, bone, kbone, joint_motion, bone_motion, kbone_motion], [2, 2, 2, 1, 1, 1])
# print('Top-1', top1(fused, label))