(pred_dict, normalization_dict)
| 100 | return gt_dict_out |
| 101 | |
| 102 | def align_pred_with_gt(pred_dict, normalization_dict): |
| 103 | #the positions (originating from integration of directions) from the codec start at 0 but we need to shift this by the amount that normalize_gt_data shifts the origin |
| 104 | strands_data=pred_dict["strand_positions"] |
| 105 | strands_data=strands_data-normalization_dict["xyz_mean"].view(1,1,3)/normalization_dict["xyz_std"].view(1,1,3) |
| 106 | pred_dict_out = dict(pred_dict) |
| 107 | pred_dict_out["strand_positions"]=strands_data |
| 108 | return pred_dict_out |
| 109 | |
| 110 | |
| 111 |
nothing calls this directly
no outgoing calls
no test coverage detected