您最多选择25个主题 主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

66 行
1.9 KiB

using System;
namespace UnityEngine.Perception.GroundTruth.Exporters.Coco
{
public static class AnnotationHandler
{
public static object HandleAnnotation(AsyncAnnotation asyncAnnotation, object annotation)
{
switch (annotation)
{
case BoundingBox2DLabeler.BoundingBoxValue bbox:
return CocoTypes.ObjectDetectionAnnotation.FromBoundingBoxValue(bbox);
case KeypointLabeler.KeypointEntry keypoint:
return CocoTypes.KeypointAnnotation.FromKeypointValue(keypoint);
}
return null;
}
public static CocoTypes.KeypointCategory ToKeypointCategory(KeypointLabeler.KeypointJson keypointJson)
{
var keypoints = new string[keypointJson.key_points.Length];
var skeleton = new int[keypointJson.skeleton.Length][];
foreach (var kp in keypointJson.key_points)
{
keypoints[kp.index] = kp.label;
}
var i = 0;
foreach (var bone in keypointJson.skeleton)
{
var joints = new int[]
{
bone.joint1,
bone.joint2
};
skeleton[i++] = joints;
}
return new CocoTypes.KeypointCategory
{
keypoints = keypoints,
skeleton = skeleton
};
}
public static object HandleCameraCapture(int id, int width, int height, string filename)
{
var image = new CocoTypes.Image()
{
id = id,
width = width,
height = height,
file_name = filename,
license = 0,
flickr_url = "",
coco_url = "",
date_captured = DateTime.Today.ToString("D")
};
return image;
}
}
}