|
|
from mmcv import load |
|
|
import sys |
|
|
|
|
|
sys.path.append('../..') |
|
|
from aclnet.smp import * |
|
|
|
|
|
|
|
|
j_1 = load('j_1/best_pred.pkl') |
|
|
b_1 = load('b_1/best_pred.pkl') |
|
|
k_1 = load('k_1/best_pred.pkl') |
|
|
jm = load('jm/best_pred.pkl') |
|
|
bm = load('bm/best_pred.pkl') |
|
|
km = load('km/best_pred.pkl') |
|
|
j_2 = load('j_2/best_pred.pkl') |
|
|
b_2 = load('b_2/best_pred.pkl') |
|
|
k_2 = load('k_2/best_pred.pkl') |
|
|
j_3 = load('j_3/best_pred.pkl') |
|
|
b_3 = load('b_3/best_pred.pkl') |
|
|
k_3 = load('k_3/best_pred.pkl') |
|
|
label = load_label('/data/finegym/gym_hrnet.pkl', 'val') |
|
|
|
|
|
|
|
|
""" |
|
|
*************** |
|
|
InfoGCN v0: |
|
|
j jm b bm k km |
|
|
2S: 95.48 |
|
|
4S: 95.80 |
|
|
6S: 96.01 |
|
|
*************** |
|
|
""" |
|
|
print('InfoGCN v0:') |
|
|
print('j jm b bm k km') |
|
|
print('2S') |
|
|
fused = comb([j_1, b_1], [1, 1]) |
|
|
print('Top-1', top1(fused, label)) |
|
|
|
|
|
print('4S') |
|
|
fused = comb([j_1, b_1, jm, bm], [9, 9, 5, 5]) |
|
|
print('Top-1', top1(fused, label)) |
|
|
|
|
|
print('6S') |
|
|
fused = comb([j_1, b_1, k_1, jm, bm, km], [9, 9, 9, 5, 5, 5]) |
|
|
print('Top-1', top1(fused, label)) |
|
|
|
|
|
|
|
|
""" |
|
|
*************** |
|
|
HD-GCN v1: |
|
|
j b j b j b |
|
|
2S: 95.48 |
|
|
4S: 95.79 |
|
|
6S: 95.93 |
|
|
*************** |
|
|
""" |
|
|
print('HD-GCN v1:') |
|
|
print('j b j b j b') |
|
|
print('2S') |
|
|
fused = comb([j_1, b_1], [1, 1]) |
|
|
print('Top-1', top1(fused, label)) |
|
|
|
|
|
print('4S') |
|
|
fused = comb([j_1, b_1, j_2, b_2], [4, 4, 3, 3]) |
|
|
print('Top-1', top1(fused, label)) |
|
|
|
|
|
print('6S') |
|
|
fused = comb([j_1, b_1, j_2, b_2, j_3, b_3], [4, 4, 3, 3, 8, 8]) |
|
|
print('Top-1', top1(fused, label)) |
|
|
|
