99 lines
4.4 KiB
C#
99 lines
4.4 KiB
C#
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
|
||
//
|
||
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
|
||
//
|
||
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
|
||
|
||
using Furion.DependencyInjection;
|
||
using Microsoft.AspNetCore.Http;
|
||
using PaddleOCRSharp;
|
||
using System.ComponentModel.DataAnnotations;
|
||
|
||
namespace Admin.NET.Plugin.PaddleOCR.Service;
|
||
|
||
/// <summary>
|
||
/// PaddleOCR 图像识别服务 🧩
|
||
/// </summary>
|
||
[ApiDescriptionSettings(PaddleOCRConst.GroupName, Description = "PaddleOCR 图像识别")]
|
||
public class PaddleOCRService : IDynamicApiController, ISingleton
|
||
{
|
||
private readonly PaddleOCREngine _engine;
|
||
private readonly PaddleStructureEngine _structengine;
|
||
|
||
public PaddleOCRService()
|
||
{
|
||
// 自带轻量版中英文模型PP-OCRv4
|
||
OCRModelConfig config = null;
|
||
|
||
//// 服务器中英文模型v2
|
||
//OCRModelConfig config = new OCRModelConfig();
|
||
//string modelPathroot = "你的模型绝对路径文件夹";
|
||
//config.det_infer = modelPathroot + @"\ch_ppocr_server_v2.0_det_infer";
|
||
//config.cls_infer = modelPathroot + @"\ch_ppocr_mobile_v2.0_cls_infer";
|
||
//config.rec_infer = modelPathroot + @"\ch_ppocr_server_v2.0_rec_infer";
|
||
//config.keys = modelPathroot + @"\ppocr_keys.txt";
|
||
|
||
//// 英文和数字模型v3
|
||
//OCRModelConfig config = new OCRModelConfig();
|
||
//string modelPathroot = "你的模型绝对路径文件夹";
|
||
//config.det_infer = modelPathroot + @"\en_PP-OCRv3_det_infer";
|
||
//config.cls_infer = modelPathroot + @"\ch_ppocr_mobile_v2.0_cls_infer";
|
||
//config.rec_infer = modelPathroot + @"\en_PP-OCRv3_rec_infer";
|
||
//config.keys = modelPathroot + @"\en_dict.txt";
|
||
|
||
//// 中英文模型V4
|
||
//config = new OCRModelConfig();
|
||
//string modelPathroot = AppContext.BaseDirectory + "OcrModel\\ch_PP-OCRv4";
|
||
//config.det_infer = modelPathroot + @"\ch_PP-OCRv4_det_infer";
|
||
//config.cls_infer = modelPathroot + @"\ch_ppocr_mobile_v2.0_cls_infer";
|
||
//config.rec_infer = modelPathroot + @"\ch_PP-OCRv4_rec_infer";
|
||
//config.keys = modelPathroot + @"\ppocr_keys.txt";
|
||
|
||
//// 服务器中英文模型V4
|
||
//config = new OCRModelConfig();
|
||
//string modelPathroot = "你的模型绝对路径文件夹";
|
||
//config.det_infer = modelPathroot + @"\ch_PP-OCRv4_det_server_infer";
|
||
//config.cls_infer = modelPathroot + @"\ch_ppocr_mobile_v2.0_cls_infer";
|
||
//config.rec_infer = modelPathroot + @"\ch_PP-OCRv4_rec_server_infer";
|
||
//config.keys = modelPathroot + @"\ppocr_keys.txt";
|
||
|
||
//// 参数
|
||
//OCRParameter oCRParameter = new OCRParameter();
|
||
// oCRParameter.use_gpu=true; // 当使用GPU版本的预测库时，该参数打开才有效果
|
||
// oCRParameter.enable_mkldnn = false;
|
||
|
||
// 初始化OCR引擎
|
||
_engine = new PaddleOCREngine(config, "");
|
||
|
||
// 模型配置，使用默认值
|
||
StructureModelConfig structureModelConfig = null;
|
||
// 表格识别参数配置，使用默认值
|
||
StructureParameter structureParameter = new();
|
||
_structengine = new PaddleStructureEngine(structureModelConfig, structureParameter);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 识别身份证 🔖
|
||
/// </summary>
|
||
/// <param name="file"></param>
|
||
/// <returns></returns>
|
||
[AllowAnonymous]
|
||
[DisplayName("识别身份证")]
|
||
public async Task<dynamic> IDCardOCR([Required] IFormFile file)
|
||
{
|
||
using var memoryStream = new MemoryStream();
|
||
await file.CopyToAsync(memoryStream);
|
||
var ocrRes = _engine.DetectText(memoryStream.ToArray());
|
||
|
||
List<TextBlock> textBlocks = ocrRes.TextBlocks;
|
||
var cardName = TextBlockUtil.ReadIdCardName(textBlocks);
|
||
var cardNo = TextBlockUtil.ReadIdCardNo(textBlocks);
|
||
var cardAddress = TextBlockUtil.ReadIdCardAddress(textBlocks);
|
||
return await Task.FromResult(new
|
||
{
|
||
CardName = cardName,
|
||
CardNo = cardNo,
|
||
CardAddress = cardAddress
|
||
});
|
||
}
|
||
} |