143 lines
8.0 KiB
C#
143 lines
8.0 KiB
C#
using OpenCVCompact;
|
||
using UnityEngine;
|
||
|
||
public class PictureUtility
|
||
{
|
||
private static Material _flipMaterial;
|
||
|
||
public static Material FlipMaterial
|
||
{
|
||
get
|
||
{
|
||
if (_flipMaterial == null)
|
||
_flipMaterial = Resources.Load<Material>("Materials/Unlit_FlipHorizontal");
|
||
return _flipMaterial;
|
||
}
|
||
|
||
set => _flipMaterial = value;
|
||
}
|
||
|
||
public static Texture2D HorizontalFlipTexture(Texture2D texture)
|
||
{
|
||
//得到图片的宽高
|
||
Texture2D retVal = new Texture2D(texture.width, texture.height);
|
||
RenderTexture tmp = RenderTexture.GetTemporary(texture.width, texture.height, 0, RenderTextureFormat.ARGB32);
|
||
Graphics.Blit(texture, tmp, _flipMaterial);
|
||
RenderTexture.active = tmp;
|
||
retVal.ReadPixels(new UnityEngine.Rect(0, 0, texture.width, texture.height), 0, 0);
|
||
RenderTexture.ReleaseTemporary(tmp);
|
||
return retVal;
|
||
}
|
||
|
||
public static OpenCVForUnity.CoreModule.Mat HorizontalFlipMat(OpenCVForUnity.CoreModule.Mat mat)
|
||
{
|
||
OpenCVForUnity.CoreModule.Mat dst = new OpenCVForUnity.CoreModule.Mat();
|
||
OpenCVForUnity.CoreModule.Core.flip(mat, dst, 1);
|
||
return dst;
|
||
}
|
||
|
||
//纠正鱼眼透视畸变
|
||
public static OpenCVForUnity.CoreModule.Mat Undistort(OpenCVForUnity.CoreModule.Mat src)
|
||
{
|
||
//摄像机的4个畸变系数：k1,k2,k3,k4
|
||
OpenCVForUnity.CoreModule.Mat K = new OpenCVForUnity.CoreModule.Mat(3, 3, OpenCVForUnity.CoreModule.CvType.CV_64FC1);
|
||
K.put(0, 0, 1151.7, 0, 985.6, 0, 1153.4, 573.004, 0, 0, 1);
|
||
|
||
OpenCVForUnity.CoreModule.Mat D = new OpenCVForUnity.CoreModule.Mat(5, 1, OpenCVForUnity.CoreModule.CvType.CV_64FC1);
|
||
//D.put(0, 0, -0.42, 0.2196, 0.0038, 0.013, -0.06);
|
||
D.put(0, 0, -0.01, 0.09, 0, 0, -0.19);
|
||
|
||
OpenCVForUnity.CoreModule.Mat retVal = src.clone();
|
||
OpenCVForUnity.Calib3dModule.Calib3d.undistort(src, retVal, K, D);
|
||
|
||
return retVal;
|
||
}
|
||
|
||
public static OpenCVCompact.Mat GetROIRect(OpenCVCompact.Mat bgrMat, out int[] roiRegion)
|
||
{
|
||
OpenCVCompact.Rect roiRect;
|
||
roiRegion = GetPositionRect(bgrMat.width(), bgrMat.height());
|
||
roiRect = new OpenCVCompact.Rect(
|
||
(int)roiRegion[0],
|
||
(int)roiRegion[1],
|
||
Mathf.Abs((int)(roiRegion[2] - roiRegion[0])),
|
||
Mathf.Abs((int)(roiRegion[3] - roiRegion[1])));
|
||
OpenCVCompact.Mat personRectImg = new OpenCVCompact.Mat(bgrMat, roiRect);
|
||
return personRectImg;
|
||
}
|
||
|
||
public static int[] GetPositionRect(int width, int height)
|
||
{
|
||
int[] retVal = new int[4];
|
||
int shift = (int)(width / 2 * 0.2f);
|
||
if (!GlobalData.Instance.IsDriverPosition) //副驾驶
|
||
{
|
||
retVal[0] = 0; //left
|
||
retVal[1] = 0; //top
|
||
retVal[2] = (width / 2) - shift; //right
|
||
retVal[3] = height; //bottom
|
||
}
|
||
else
|
||
{
|
||
retVal[0] = (width / 2) + shift; //left
|
||
retVal[1] = 0; //top
|
||
retVal[2] = width; //right
|
||
retVal[3] = height; //bottom
|
||
}
|
||
return retVal;
|
||
}
|
||
|
||
public static void DrawAimBox(OpenCVForUnity.CoreModule.Mat img, float[] box, float shift, int lineLength)
|
||
{
|
||
float left = box[0] + shift;
|
||
float top = box[1] + shift;
|
||
float right = box[2] - shift;
|
||
float bottom = box[3] - shift;
|
||
|
||
OpenCVForUnity.ImgprocModule.Imgproc.cvtColor(img, img, OpenCVForUnity.ImgprocModule.Imgproc.COLOR_RGBA2RGB);
|
||
|
||
//界面蓝 RGB = 19, 115, 230
|
||
OpenCVForUnity.ImgprocModule.Imgproc.rectangle(img, new OpenCVForUnity.CoreModule.Point(left, top), new OpenCVForUnity.CoreModule.Point(right, bottom), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 50), 2, 3);
|
||
//右下角
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(right, bottom), new OpenCVForUnity.CoreModule.Point(right - lineLength, bottom), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(right, bottom), new OpenCVForUnity.CoreModule.Point(right, bottom - lineLength), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
//左下角
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(left, bottom), new OpenCVForUnity.CoreModule.Point(left + lineLength, bottom), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(left, bottom), new OpenCVForUnity.CoreModule.Point(left, bottom - lineLength), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
//右上角
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(right, top), new OpenCVForUnity.CoreModule.Point(right - lineLength, top), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(right, top), new OpenCVForUnity.CoreModule.Point(right, top + lineLength), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
//左上角
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(left, top), new OpenCVForUnity.CoreModule.Point(left + lineLength, top), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVForUnity.ImgprocModule.Imgproc.line(img, new OpenCVForUnity.CoreModule.Point(left, top), new OpenCVForUnity.CoreModule.Point(left, top + lineLength), new OpenCVForUnity.CoreModule.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVForUnity.ImgprocModule.Imgproc.cvtColor(img, img, OpenCVForUnity.ImgprocModule.Imgproc.COLOR_RGB2RGBA);
|
||
}
|
||
|
||
public static void DrawAimBoxFaceDetect(OpenCVCompact.Mat img, float[] box, float shift, int lineLength)
|
||
{
|
||
float left = box[0] + shift;
|
||
float top = box[1] + shift;
|
||
float right = box[2] - shift;
|
||
float bottom = box[3] - shift;
|
||
|
||
OpenCVCompact.Imgproc.cvtColor(img, img, OpenCVCompact.Imgproc.COLOR_RGBA2RGB);
|
||
|
||
//界面蓝 RGB = 19, 115, 230
|
||
OpenCVCompact.Imgproc.rectangle(img, new OpenCVCompact.Point(left, top), new OpenCVCompact.Point(right, bottom), new OpenCVCompact.Scalar(255, 255, 255, 50), 2, 3);
|
||
//右下角
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(right, bottom), new OpenCVCompact.Point(right - lineLength, bottom), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(right, bottom), new OpenCVCompact.Point(right, bottom - lineLength), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
//左下角
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(left, bottom), new OpenCVCompact.Point(left + lineLength, bottom), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(left, bottom), new OpenCVCompact.Point(left, bottom - lineLength), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
//右上角
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(right, top), new OpenCVCompact.Point(right - lineLength, top), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(right, top), new OpenCVCompact.Point(right, top + lineLength), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
//左上角
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(left, top), new OpenCVCompact.Point(left + lineLength, top), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVCompact.Imgproc.line(img, new OpenCVCompact.Point(left, top), new OpenCVCompact.Point(left, top + lineLength), new OpenCVCompact.Scalar(255, 255, 255, 255), 10, 3);
|
||
OpenCVCompact.Imgproc.cvtColor(img, img, OpenCVCompact.Imgproc.COLOR_RGB2RGBA);
|
||
}
|
||
}
|
||
|