效果
项目
代码
using OpenCvSharp;
using OpenCvSharp.Extensions;
using Sdcb.PaddleDetection;
using Sdcb.PaddleInference;
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Windows.Forms;
namespace PaddleDetection_版面分析
{
public partial class Form1 : Form
{
public Form1()
{
InitializeComponent();
}
Bitmap bmp;
string fileFilter = "*.*|*.bmp;*.jpg;*.jpeg;*.tiff;*.tiff;*.png";
string img = "";
double fontScale = 2D;
int thickness = 4;
LineTypes lineType = LineTypes.Link4;
PaddleConfig paddleConfig;
PaddleDetector d;
String startupPath;
float confidence = 0.75f;
DetectionResult[] r;
Mat src;
private void button1_Click(object sender, EventArgs e)
{
OpenFileDialog ofd = new OpenFileDialog();
ofd.Filter = fileFilter;
if (ofd.ShowDialog() != DialogResult.OK) return;
pictureBox1.Image = null;
img = ofd.FileName;
bmp = new Bitmap(img);
pictureBox1.Image = new Bitmap(img);
src = Cv2.ImRead(img);
}
private void Form1_Load(object sender, EventArgs e)
{
startupPath = System.Windows.Forms.Application.StartupPath;
paddleConfig = PaddleConfig.FromModelDir(startupPath + "\\ppyolov2_r50vd_dcn_365e_publaynet\\");
string configYmlPath = startupPath + "\\ppyolov2_r50vd_dcn_365e_publaynet\\infer_cfg.yml";
d = new PaddleDetector(paddleConfig, configYmlPath);
}
private void button2_Click(object sender, EventArgs e)
{
if (pictureBox1.Image == null)
{
return;
}
r = d.Run(src);
for (int i = 0; i < r.Length; i++)
{
if (r[i].Confidence > confidence)
{
Scalar scalar = Scalar.RandomColor();
Cv2.Rectangle(src, r[i].Rect, scalar, 1, LineTypes.Link8, 0);
Cv2.PutText(src, r[i].LabelName, new OpenCvSharp.Point(r[i].Rect.X + r[i].Rect.Width / 2, r[i].Rect.Y + r[i].Rect.Height / 2), HersheyFonts.HersheyComplex, fontScale, scalar, thickness, lineType, false);
}
}
pictureBox2.Image = BitmapConverter.ToBitmap(src);
}
}
}
Demo下载