Convert predictions to deeplabcut assemblies.
(eval_results_obj, keypoint_names)
| 116 | |
| 117 | |
| 118 | def conv_obj_to_assemblies(eval_results_obj, keypoint_names): |
| 119 | """Convert predictions to deeplabcut assemblies.""" |
| 120 | assemblies = {} |
| 121 | for image_path, results in eval_results_obj.items(): |
| 122 | lst = [] |
| 123 | for dict_ in results: |
| 124 | ass = inferenceutils.Assembly(len(keypoint_names)) |
| 125 | for i, kpt in enumerate(keypoint_names): |
| 126 | xy = dict_["pose"][kpt] |
| 127 | if ~np.isnan(xy).all(): |
| 128 | joint = inferenceutils.Joint(pos=(xy), label=i) |
| 129 | ass.add_joint(joint) |
| 130 | # TODO(jeylau) add affinity.setter to Assembly |
| 131 | ass._affinity = dict_["score"] |
| 132 | ass._links = [None] |
| 133 | if len(ass): |
| 134 | lst.append(ass) |
| 135 | assemblies[image_path] = lst |
| 136 | return assemblies |
| 137 | |
| 138 | |
| 139 | def calc_map_from_obj( |
no test coverage detected