查看原文
其他

在Unity中实现手部跟踪

Jiasheng Tang Unity官方平台 2022-05-07

本文将由游戏开发者Jiasheng Tang分享在Unity中实现手部跟踪的三种方法。


很多人小时候,一直梦想着使用双手来远程操控物体。通常,我们可以触碰、移动、滚动和投掷物体,但我们不能像电影中的巫师或绝地大师,可以在不接触物体的情况下操控物体。



虽然这个梦想在现实生活中难以企及,但在虚拟世界中是存在可能的。


Microsoft Kinect曾非常受欢迎,因为用户可以通过使用它,来用双手切掉虚拟的水果。Leap Motion允许用户以精妙的方式和虚拟物体进行交互。


这些产品都需要使用额外的硬件,费用比较昂贵。即使拥有相应设备,普通用户也较难实现这种体验。如果有一种方法只使用带有普通RGB摄像头智能手机,就可以跟踪双手,这会是多酷的体验。


本文将介绍如何在Unity中通过使用RGB摄像头实现手部跟踪。


学习准备

学习本文,你需要掌握以下知识内容:

  • Unity的相关知识,本文开发使用的是Unity 2018.3.14。


  • OpenCV的基础知识。你需要在Asset Store资源商店获取OpenCV For Unity插件:

    https://assetstore.unity.com/packages/tools/integration/opencv-for-unity-21088


  • 对神经网络的基本了解。


手部跟踪的方法总结


使用RGB进行手部跟踪有三种方法:Haar级联方法,轮廓方法和神经网络方法。


下面是对三种方法的总结:

Haar级联方法

简单来说,这种方法很容易实现,而且速度非常快,但缺点是非常不稳定。


Haar级联方法是跟踪面部的常用方法,但手和脸不同,手没有固定的形态。如果只希望找到静态的标准手部图像,例如:手掌向前的张手画面,这种方法可能会起到作用。


轮廓方法

这是一种直截了当并易于实现的方法,提供很多可以定制的参数来根据用例调整,而且在计算方面的开销也不是很大。


如果用户在自己房间使用该方法,它会有很好的效果。但在移动设备上,由于光线会发生变化,背景会移动,而且用户周围会有其他人,因此该方法没有很理想的效果。


神经网络方法

这种方法在三种方法中有最强的稳定性,可以应对多种情形。然而,这种方法在计算方面的开销很大。


手部跟踪的实现

下面将介绍相应的代码,你可以访问GitHub获取完整的代码。

https://github.com/teejs2012/Hand_Tracking_Unity3D


Haar级联方法

Haar级联是使用机器学习提取特征到XML文件的方法。


首先,我们需要获取训练好的XML文件,该文件叫palm.xml,它专门为识别手掌而训练。


获取训练好的XML文件,请访问:

https://github.com/Balaje/OpenCV/blob/master/haarcascades/palm.xml


在下载好该文件后,把它移动到Unity中的StreamingAssets文件夹。


加载级联文件时,使用以下代码: 

var cascadeFileName = Utils.getFilePath("palm.xml");

cascadeDetector = new CascadeClassifier();

cascadeDetector.load(cascadeFileName);


运行检测时,我们会进行一些必要的预处理过程:把图像转换为灰度图,并调整直方图。然后,我们可以调用detectMultiScale函数。

MatOfRect hands = new MatOfRect();


Mat gray = new Mat(imgHeight, imgWidth, CvType.CV_8UC3);


Imgproc.cvtColor(image, gray, Imgproc.COLOR_BGR2GRAY);


Imgproc.equalizeHist(gray, gray);


cascadeDetector.detectMultiScale(gray, hands, 1.1, 2, 0 | Objdetect.CASCADE_DO_CANNY_PRUNING | Objdetect.CASCADE_SCALE_IMAGE | Objdetect.CASCADE_FIND_BIGGEST_OBJECT, new Size(10, 10), new Size());


OpenCVForUnity.CoreModule.Rect[] handsArray = hands.toArray();

if (handsArray.Length != 0){

//手部已经检测到

}


轮廓方法

基于轮廓的方法是很直接的方法,这种方法只使用了计算机视觉,没有使用比较特别的模型。


轮廓方法基本上有二个主要步骤:

  • 找到图像上符合人类皮肤颜色的区域。

  • 找到符合手指的轮廓形状。


人类皮肤颜色通常是色谱的一部分,为了找到这类颜色,我们需要把颜色形式从RGB转换为YCrCb,并检查每个像素是否都在该范围内。

Mat YCrCb_image = new Mat();

int Y_channel = 0;

int Cr_channel = 1;

int Cb_channel = 2;


Imgproc.cvtColor(imgMat, YCrCb_image, Imgproc.COLOR_RGB2YCrCb);

var output_mask = Mat.zeros(imgWidth, imgHeight, CvType.CV_8UC1);


for (int i = 0; i < YCrCb_image.rows(); i++)

{

for (int j = 0; j < YCrCb_image.cols(); j++)

{

double[] p_src = YCrCb_image.get(i, j);

if (p_src[Y_channel] > 80 && p_src[Cr_channel] > 135 && p_src[Cr_channel] < 180 && p_src[Cb_channel] > 85 && p_src[Cb_channel] < 135)

{

output_mask.put(i, j, 255);

}

}


最终结果是一个遮罩,符合人类皮肤颜色的像素是白色,剩余部分是黑色。


在获得这个遮罩后,我们会提取遮罩的凸壳和轮廓,并检测“瑕疵”。“瑕疵”点表示轮廓线上远离凸壳的点。如果“瑕疵”的角度和长度等参数符合特定标准,那么我们就知道“瑕疵”对应着手指。


在找到足够的“手指瑕疵”时,我们会告诉系统找到了手部,根据经验,我们把这里的“足够”定义为1和4之间的数量。


下图是瑕疵点的图示,蓝线是凸壳,绿线是轮廓。


//在图像中找到轮廓

List<MatOfPoint> contours = new List<MatOfPoint>();

Imgproc.findContours(maskImage, contours, new MatOfPoint(), Imgproc.RETR_EXTERNAL, Imgproc.CHAIN_APPROX_SIMPLE);


var points = new MatOfPoint(contours[index].toArray());

var hull = new MatOfInt();

Imgproc.convexHull(points, hull, false);


//找到瑕疵

var defects = new MatOfInt4();

Imgproc.convexityDefects(points, hull, defects);

var start_points = new MatOfPoint2f();

var far_points = new MatOfPoint2f();


//循环检查瑕疵,了解它是否符合条件

for (int i = 0; i < defects.size().height; i++)

{

int ind_start = (int)defects.get(i, 0)[0];

int ind_end = (int)defects.get(i, 0)[1];

int ind_far = (int)defects.get(i, 0)[2];

double depth = defects.get(i, 0)[3] / 256;

double a = Core.norm(contours[index].row(ind_start) - contours[index].row(ind_end));

double b = Core.norm(contours[index].row(ind_far) - contours[index].row(ind_start));

double c = Core.norm(contours[index].row(ind_far) - contours[index].row(ind_end));

double angle = Math.Acos((b * b + c * c - a * a) / (2 * b * c)) * 180.0 / Math.PI;

double threshFingerLength = ((double)maskImage.height()) / 8.0;

double threshAngle = 80;

if (angle < threshAngle && depth > threshFingerLength)

{


//起点

var aa = contours[index].row(ind_start);

start_points.push_back(contours[index].row(ind_start));

far_points.push_back(contours[index].row(ind_far));

}

}


// 检查找到的瑕疵是否在范围内

if (far_points.size().height > min_defects_count && far_points.size().height < max_defects_count)

{

//检测到手部

}


神经网络方式

在近几年,与用于许多任务的传统解决方案相比,基于神经网络的解决方案实现了更好的性能,这在计算机视觉领域特别明显。因为我们的任务是计算机视觉任务,所以我们自然希望使用神经网络。


实际上,目标检测在GitHub上有许多关于手部跟踪的优秀项目。例如:Victordibia的handtracking项目使用经典的目标检测模型架构 - 单步检测SSD。


下载了解Victordibia的handtracking项目:

https://github.com/victordibia/handtracking


单步检测SSD因其快捷的推理速度而出名,适用于实时应用。为了进一步提升速度,该模型使用了MobileNet结构来进行调整,最后我们得到了SSDMobileNet结构,预训练模型的大小只有约20MB的大小。


然而,这些解决方案使用Python和TensorFlow框架依赖来编写,我们要如何把它使用到Unity中呢?


一个热门方法是使用TensorflowSharp,即TensorFlow框架的C#版本。但这种方法的速度对实时应用来说太慢了,帧率甚至不到1fps。


我选择使用了OpenCV的DNN模块。该模块是OpenCVForUnity插件的一部分,提供了一些示例场景来展示使用方法。如果想要使用自定义模型,需要执行额外的步骤来把模型转换为OpenCV理解的格式。



首先要转换模型。OpenCV需要特别的pbtxt文件来使模型正常加载。生成该文件时,我们需要模型的冻结视图,以及针对模型的管线配置。


我从这个GitHub项目获取了二个相应的文件frozen_inference_graph.pb和ssd_mobilenet_v1_coco.config。


转换过程的实现,请查看Github的代码,生成出来的frozen_inference_graph.pbtxt文件也包含在其项目中。

https://github.com/teejs2012/OpenCV_DNN_with_Tensorflow_model


在准备好frozen_inference_graph.pb和frozen_inference_graph.pbtxt文件后,我们要把它们移动到Unity中的StreamingAssets文件夹。


加载模型时,使用以下代码:

var modelPath = Utils.getFilePath("frozen_inference_graph.pb");

var configPath = Utils.getFilePath("frozen_inference_graph.pbtxt");

tfDetector = Dnn.readNetFromTensorflow(modelPath, configPath);


进行检测时,使用以下代码:

var blob = Dnn.blobFromImage(image, 1, new Size(300, 300), new Scalar(0, 0, 0), true, false);

tfDetector.setInput(blob);

Mat prob = tfDetector.forward();

Mat newMat = prob.reshape(1, (int)prob.total() / prob.size(3));


float maxScore = 0;

int scoreInd = 0;


for (int i = 0; i < newMat.rows(); i++)

{

var score = (float)newMat.get(i, 2)[0];

if (score > maxScore)

{

maxScore = score;

scoreInd = i;

}

}

if (maxScore > thresholdScore)

{

// 检测到手部

}


我们要注意的一个参数是blobFromImage函数中的Size值。在传递到模型前,图像会调整为该数值的大小。


建议把Size设为300,因为这是模型的训练图像的大小,但如果帧率对应用很重要,该数值可以减少为150。


手指跟踪

你可能会注意到,我们做的这些事情有一个很大的限制:我们只能跟踪“手部”,而手指跟踪不包含在工具包中。


跟踪手指的方法也很多,其中一个流行方法是使用神经网络模型从RGB图像中估算3D手部姿势,你可以参考和下载以下代码:

https://github.com/Hzzone/pytorch-openpose


小结

在Unity中实现手部跟踪的方法为大家介绍到这里,让我们行动起来,在虚拟世界中成为可隔空控物的绝地大师吧。


下载Unity Connect APP,请点击此处 观看部分Unity官方视频,请关注B站帐户:Unity官方


你可以访问Unity答疑专区留下你的问题,Unity社区和官方团队帮你解答:

Connect.unity.com/g/discussion


推荐阅读

配置Unity机器学习代理工具和TensorFlow环境(Windows 10)

使用Unity ML-Agents Toolkit训练的可爱柯基犬Puppo

Obstacle Tower挑战赛获奖者以及发布开源环境

使用Unity 2019.2最新地形工具加速地形创作

使用JetBrains Rider实现更快的C#编程

Unite 2019|Unity的光照烘焙技术


官方活动

成为Unity Buddy,享受专属福利

8月21日前加入Unity Connect,创建个人频道,成为与Unity社区同行的伙伴-Unity Buddy,享受专属福利。


Unity Connect评论区支持你认可的Buddy,有机会获得Buddy送出的Unity周边。

了解如何成为Unity Buddy,请点击此处



喜欢本文,点击“在看”

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存