未验证 提交 190e237b 编写于 作者: J JYChen 提交者: GitHub

fix box filter when box_num > 0 but with no target class (#6506)

上级 324b0b99
...@@ -37,8 +37,15 @@ KEYPOINT_SUPPORT_MODELS = { ...@@ -37,8 +37,15 @@ KEYPOINT_SUPPORT_MODELS = {
def predict_with_given_det(image, det_res, keypoint_detector, def predict_with_given_det(image, det_res, keypoint_detector,
keypoint_batch_size, run_benchmark): keypoint_batch_size, run_benchmark):
keypoint_res = {}
rec_images, records, det_rects = keypoint_detector.get_person_from_rect( rec_images, records, det_rects = keypoint_detector.get_person_from_rect(
image, det_res) image, det_res)
if len(det_rects) == 0:
keypoint_res['keypoint'] = [[], []]
return keypoint_res
keypoint_vector = [] keypoint_vector = []
score_vector = [] score_vector = []
...@@ -47,7 +54,6 @@ def predict_with_given_det(image, det_res, keypoint_detector, ...@@ -47,7 +54,6 @@ def predict_with_given_det(image, det_res, keypoint_detector,
rec_images, run_benchmark, repeats=10, visual=False) rec_images, run_benchmark, repeats=10, visual=False)
keypoint_vector, score_vector = translate_to_ori_images(keypoint_results, keypoint_vector, score_vector = translate_to_ori_images(keypoint_results,
np.array(records)) np.array(records))
keypoint_res = {}
keypoint_res['keypoint'] = [ keypoint_res['keypoint'] = [
keypoint_vector.tolist(), score_vector.tolist() keypoint_vector.tolist(), score_vector.tolist()
] if len(keypoint_vector) > 0 else [[], []] ] if len(keypoint_vector) > 0 else [[], []]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册