// 显示
private static void Display(EndPointDectector dectector, Mat frame, PointF crossPoint)
{
//绘制roi跟跟踪框
CvInvoke.Rectangle(dectector.FrameScale, dectector.Roi, new MCvScalar(255, 0, 0), 2, LineType.FourConnected);
//绘制处理后的图像的ROI区域
CvInvoke.Imshow("scale tracker", dectector.FrameScale);
//绘制原图的跟踪框
CvInvoke.Rectangle(dectector.Frame, dectector.RoiScale, new MCvScalar(255, 0, 0), 2); //绘制原图的ROI区域
CvInvoke.Imshow("input tracker", frame);
// KMeans
CvInvoke.Imshow("kmeans", dectector.Roi_Kmeans);
//// sobel
//CvInvoke.Imshow("edge_sobel", dectector.RoiSobel);
//canny
//CvInvoke.Imshow("Canny", dectector.RoiCanny);
if (dectector.Lines.Length > 1) //判断检测的直线条数，必须要大于等于2
{
CvInvoke.Line(dectector.RoiTemp, dectector.FinalLinestand.P1, dectector.FinalLinestand.P2,
new MCvScalar(0, 0, 255), 2, LineType.EightConnected);
CvInvoke.Line(dectector.RoiTemp, dectector.FinalLineanother.P1, dectector.FinalLineanother.P2,
new MCvScalar(0, 0, 255), 2, LineType.EightConnected);
CvInvoke.Circle(dectector.RoiSingletemp, new Point((int)crossPoint.X, (int)crossPoint.Y), 3,
new MCvScalar(255, 100, 0));
}
else if (dectector.Lines.Length == 1)
{
Console.WriteLine("只检测了一条直线！");
}
else
{
Console.WriteLine("检测不到直线！");
}
CvInvoke.Imshow("tracker houghline", dectector.RoiTemp); //显示末端点的位置
CvInvoke.Imshow("roi_singletemp", dectector.RoiSingletemp); //显示末端点的位置
}
static int Main(string[] args)
{
//可选BOOSTING, MIL, KCF, TLD, MEDIANFLOW, or GOTURN
Rectangle roi = new Rectangle();
Mat frame = new Mat();
Mat frame_scale = new Mat();
//读入视频
string video = "S-dst.avi"; //S-800v1.avi
VideoCapture cap = new VideoCapture(video);
if (!cap.IsOpened)
{
Console.WriteLine("cannot read video!");
return(-1);
}
frame = cap.QueryFrame();
CvInvoke.Resize(frame, frame_scale, new Size(480, 320), 0, 0, Inter.Linear);//缩小尺寸
//设置roi区域
// roi = selectROI("scale tracker", frame_scale);
roi = new Rectangle(375 - 5, 122 - 5, 21 + 5, 58 + 5);
if (roi.Width == 0 || roi.Height == 0)
{
return(0);
}
// 位置检测器
EndPointDectector dectector = new EndPointDectector(frame_scale, roi);
Test1();
// 窗口显示
Console.WriteLine("Start the tracking process, press ESC to quit.\n");
CvInvoke.NamedWindow("scale tracker", NamedWindowType.Normal);
CvInvoke.NamedWindow("input tracker", NamedWindowType.Normal);
CvInvoke.NamedWindow("roi_singletemp", NamedWindowType.Normal);
CvInvoke.NamedWindow("tracker houghline", NamedWindowType.Normal);
while (true)
{
// 获取图像帧
frame = cap.QueryFrame();
PointF crossPoint = dectector.Detect(frame);
Console.WriteLine("检测到的坐标点：" + crossPoint.ToString());
Display(dectector, frame, crossPoint);
//循环时间设定
if (CvInvoke.WaitKey(1) == 27)
{
break;
}
}
cap.Dispose();
return(0);
}
