| 167 | return eval_result |
| 168 | |
| 169 | def print_eval_result(self, eval_result): |
| 170 | |
| 171 | print('======ARCTIC-val======') |
| 172 | print('PA MPVPE (All): %.2f mm' % np.mean(eval_result['pa_mpvpe_all'])) |
| 173 | print('PA MPVPE (L-Hands): %.2f mm' % |
| 174 | np.mean(eval_result['pa_mpvpe_l_hand'])) |
| 175 | print('PA MPVPE (R-Hands): %.2f mm' % |
| 176 | np.mean(eval_result['pa_mpvpe_r_hand'])) |
| 177 | print('PA MPVPE (Hands): %.2f mm' % |
| 178 | np.mean(eval_result['pa_mpvpe_hand'])) |
| 179 | print('PA MPVPE (Face): %.2f mm' % |
| 180 | np.mean(eval_result['pa_mpvpe_face'])) |
| 181 | print() |
| 182 | |
| 183 | print('MPVPE (All): %.2f mm' % np.mean(eval_result['mpvpe_all'])) |
| 184 | print('MPVPE (L-Hands): %.2f mm' % |
| 185 | np.mean(eval_result['mpvpe_l_hand'])) |
| 186 | print('MPVPE (R-Hands): %.2f mm' % |
| 187 | np.mean(eval_result['mpvpe_r_hand'])) |
| 188 | print('MPVPE (Hands): %.2f mm' % np.mean(eval_result['mpvpe_hand'])) |
| 189 | print('MPVPE (Face): %.2f mm' % np.mean(eval_result['mpvpe_face'])) |
| 190 | |
| 191 | out_file = osp.join(cfg.result_dir,'arctic_val.txt') |
| 192 | if os.path.exists(out_file): |
| 193 | f = open(out_file, 'a+') |
| 194 | else: |
| 195 | f = open(out_file, 'w', encoding="utf-8") |
| 196 | f.write('\n') |
| 197 | f.write(f'{cfg.exp_name}\n') |
| 198 | f.write(f'ARCTIC-val dataset: \n') |
| 199 | f.write('PA MPVPE (All): %.2f mm\n' % |
| 200 | np.mean(eval_result['pa_mpvpe_all'])) |
| 201 | f.write('PA MPVPE (L-Hands): %.2f mm\n' % |
| 202 | np.mean(eval_result['pa_mpvpe_l_hand'])) |
| 203 | f.write('PA MPVPE (R-Hands): %.2f mm\n' % |
| 204 | np.mean(eval_result['pa_mpvpe_r_hand'])) |
| 205 | f.write('PA MPVPE (Hands): %.2f mm\n' % |
| 206 | np.mean(eval_result['pa_mpvpe_hand'])) |
| 207 | f.write('PA MPVPE (Face): %.2f mm\n' % |
| 208 | np.mean(eval_result['pa_mpvpe_face'])) |
| 209 | f.write('MPVPE (All): %.2f mm\n' % np.mean(eval_result['mpvpe_all'])) |
| 210 | f.write('MPVPE (L-Hands): %.2f mm\n' % |
| 211 | np.mean(eval_result['mpvpe_l_hand'])) |
| 212 | f.write('MPVPE (R-Hands): %.2f mm\n' % |
| 213 | np.mean(eval_result['mpvpe_r_hand'])) |
| 214 | f.write('MPVPE (Hands): %.2f mm\n' % np.mean(eval_result['mpvpe_hand'])) |
| 215 | f.write('MPVPE (Face): %.2f mm\n' % np.mean(eval_result['mpvpe_face'])) |