68 lines
2.3 KiB
C#
68 lines
2.3 KiB
C#
|
using UnityEngine;
|
|||
|
|
using UnityEngine.SceneManagement;
|
|||
|
|
using System.Collections;
|
|||
|
|
using OpenCVForUnity.CoreModule;
|
|||
|
|
using OpenCVForUnity.ImgprocModule;
|
|||
|
|
using OpenCVForUnity.UnityUtils;
|
|||
|
|
|
|||
|
|
namespace OpenCVForUnityExample
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// HoughLinesP Example
|
|||
|
|
/// An example of straight line detection using the Imgproc.HoughLinesP function.
|
|||
|
|
/// http://docs.opencv.org/3.1.0/d9/db0/tutorial_hough_lines.html
|
|||
|
|
/// </summary>
|
|||
|
|
public class HoughLinesPExample : MonoBehaviour
|
|||
|
|
{
|
|||
|
|
// Use this for initialization
|
|||
|
|
void Start ()
|
|||
|
|
{
|
|||
|
|
Texture2D imgTexture = Resources.Load ("chessboard") as Texture2D;
|
|||
|
|
|
|||
|
|
Mat imgMat = new Mat (imgTexture.height, imgTexture.width, CvType.CV_8UC3);
|
|||
|
|
|
|||
|
|
Utils.texture2DToMat (imgTexture, imgMat);
|
|||
|
|
Debug.Log ("imgMat.ToString() " + imgMat.ToString ());
|
|||
|
|
|
|||
|
|
|
|||
|
|
Mat grayMat = new Mat ();
|
|||
|
|
Imgproc.cvtColor (imgMat, grayMat, Imgproc.COLOR_RGB2GRAY);
|
|||
|
|
|
|||
|
|
|
|||
|
|
Imgproc.Canny (grayMat, grayMat, 50, 200);
|
|||
|
|
|
|||
|
|
Mat lines = new Mat ();
|
|||
|
|
|
|||
|
|
Imgproc.HoughLinesP (grayMat, lines, 1, Mathf.PI / 180, 50, 50, 10);
|
|||
|
|
|
|||
|
|
// Debug.Log ("lines.toStirng() " + lines.ToString ());
|
|||
|
|
// Debug.Log ("lines.dump()" + lines.dump ());
|
|||
|
|
|
|||
|
|
int[] linesArray = new int[lines.cols () * lines.rows () * lines.channels ()];
|
|||
|
|
lines.get (0, 0, linesArray);
|
|||
|
|
|
|||
|
|
for (int i = 0; i < linesArray.Length; i = i + 4) {
|
|||
|
|
Imgproc.line (imgMat, new Point (linesArray [i + 0], linesArray [i + 1]), new Point (linesArray [i + 2], linesArray [i + 3]), new Scalar (255, 0, 0), 2);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Texture2D texture = new Texture2D (imgMat.cols (), imgMat.rows (), TextureFormat.RGBA32, false);
|
|||
|
|
Utils.matToTexture2D (imgMat, texture);
|
|||
|
|
|
|||
|
|
gameObject.GetComponent<Renderer> ().material.mainTexture = texture;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Update is called once per frame
|
|||
|
|
void Update ()
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// Raises the back button click event.
|
|||
|
|
/// </summary>
|
|||
|
|
public void OnBackButtonClick ()
|
|||
|
|
{
|
|||
|
|
SceneManager.LoadScene ("OpenCVForUnityExample");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|