copilot-swe-agent[bot]andkukulu00
ac7ca0f4d8
Agent-Logs-Url: https://github.com/IBB666/IBB_POSE/sessions/c9c0cbf1-b9de-4070-b7ee-8bda69755661 Co-authored-by: kukulu00 <1170651+kukulu00@users.noreply.github.com>
24 lines
664 B
Python
24 lines
664 B
Python
# Author: T. S. Liang @ Rama Alpaca
|
|
# Emails: tsliang2001@gmail.com | shuangliang@ramaalpaca.com
|
|
# Date: Feb. 2025
|
|
# Description: Build COCO-style keypoint metrics for SDPose evaluation.
|
|
# License: MIT License (see LICENSE for details)
|
|
|
|
|
|
from mmpose.evaluation.metrics import CocoMetric, CocoWholeBodyMetric
|
|
|
|
def get_metric(ann_file, mode = "body"):
|
|
|
|
if mode == "body":
|
|
|
|
metric = CocoMetric(iou_type='keypoints', ann_file = ann_file)
|
|
|
|
elif mode == "wholebody":
|
|
|
|
metric = CocoWholeBodyMetric(iou_type='keypoints', ann_file = ann_file)
|
|
|
|
else:
|
|
|
|
print("Error, please specify the correct type of keypoint scheme.")
|
|
|
|
return metric |