文档校正的 | 您所在的位置:网站首页 › 文档矫正功能 › 文档校正的 |
package com.huawei.mlkit.intermediate; import androidx.appcompat.app.AppCompatActivity; import android.content.Intent; import android.graphics.Bitmap; import android.graphics.BitmapFactory; import android.graphics.Point; import android.os.Bundle; import android.util.SparseArray; import android.view.View; import android.widget.Button; import android.widget.ImageView; import android.widget.Toast; import com.huawei.alliance.ml.ImageSelectedAct; import com.huawei.hmf.tasks.OnFailureListener; import com.huawei.hmf.tasks.OnSuccessListener; import com.huawei.hmf.tasks.Task; import com.huawei.hms.mlsdk.common.MLFrame; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewCorrectionAnalyzer; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewCorrectionAnalyzerFactory; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewCorrectionAnalyzerSetting; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewCorrectionConstant; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewCorrectionCoordinateInput; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewCorrectionResult; import com.huawei.hms.mlsdk.dsc.MLDocumentSkewDetectResult; import java.util.ArrayList; import java.util.List; public class MainActivity extends AppCompatActivity { // 选择的图片回调 requestCode - 请不要删除,否则影响考试成绩 private static final int REQUEST_IMG = 1000; // 选择的图片 - 请不要删除,否则影响考试成绩 private ImageView mIvSelected; // 矫正的图片 - 请不要删除,否则影响考试成绩 private ImageView mIvSkewResult; // 选择图片实体类 - 请不要删除,否则影响考试成绩 private int mImageId; //选择图片按钮 private Button mBtnImagerSelect; @Override protected void onCreate(Bundle savedInstanceState) { super.onCreate(savedInstanceState); setContentView(R.layout.activity_main); initView(); } /** * 初始化View - 请不要删除,否则影响考试成绩 */ private void initView() { // 选择的图片 - 请不要删除,否则影响考试成绩 mIvSelected = findViewById(R.id.selectedImg); // 矫正图片 - 请不要删除,否则影响考试成绩 mIvSkewResult = findViewById(R.id.documentResultImg); // TODO 获取id为btn_imageSelect的Button控件mBtnImagerSelect mBtnImagerSelect = findViewById(R.id.btn_imageSelect); if (null != mBtnImagerSelect){ mBtnImagerSelect.setOnClickListener(new View.OnClickListener() { @Override public void onClick(View v) { // 实现选择图片功能 - 请不要删除,否则影响考试成绩 startActivityForResult(new Intent(MainActivity.this, ImageSelectedAct.class), REQUEST_IMG); } }); } findViewById(R.id.btn_documentSkew).setOnClickListener(new View.OnClickListener() { @Override public void onClick(View v) { // 实现文档矫正功能 - 请不要删除,否则影响考试成绩 onDocumentSkewAsync(); } }); } /** * 展示选择图片 - 请不要删除,否则影响考试成绩 */ private void createImageBySelectedImageId() { if (mImageId == 0) { return; } mIvSelected.setImageResource(mImageId); } /** * 实现文档矫正功能 ~ 文档检测 - 请不要删除,否则影响考试成绩 */ private void onDocumentSkewAsync() { if (mImageId == 0) { Toast.makeText(this, "请选择图片", Toast.LENGTH_SHORT).show(); return; } // TODO 创建文本框检测/校正分析器。 MLDocumentSkewCorrectionAnalyzerSetting setting = new MLDocumentSkewCorrectionAnalyzerSetting.Factory().create(); MLDocumentSkewCorrectionAnalyzer analyzer = MLDocumentSkewCorrectionAnalyzerFactory.getInstance().getDocumentSkewCorrectionAnalyzer(setting); // 根据选择图片的Id生成Bitmap - 请不要删除,否则影响考试成绩 Bitmap bitmap = BitmapFactory.decodeResource(this.getResources(), mImageId); // TODO 根据选择图片的Id生成的Bitmap,创建MLFrame对象 MLFrame frame = MLFrame.fromBitmap(bitmap); // TODO 文本框检测/校正分析器 MLDocumentSkewCorrectionAnalyzer调用异步检测文档接口 asyncDocumentSkewDetect Task detectTask = analyzer.asyncDocumentSkewDetect(frame); // TODO 回调接口添加addOnSuccessListener监听,在onSuccess(MLDocumentSkewDetectResult detectResult)回调成功方法中调用 文档矫正接口 onDocumentSkewCorrect detectTask.addOnSuccessListener(new OnSuccessListener() { @Override public void onSuccess(MLDocumentSkewDetectResult detectResult) { // 检测成功。 onDocumentSkewCorrect(analyzer,frame,detectResult); } }).addOnFailureListener(new OnFailureListener() { @Override public void onFailure(Exception e) { // 检测失败。 } }); } /** * 实现文档矫正功能 ~ 矫正 - 请不要删除,否则影响考试成绩 */ private void onDocumentSkewCorrect(MLDocumentSkewCorrectionAnalyzer skewAnalyzer, MLFrame frame, MLDocumentSkewDetectResult detectResult) { if (skewAnalyzer == null || frame == null || detectResult == null) { return; } // TODO 根据detectResult 生成 左上,右上,右下,左下 四个点 Point Point leftTop = detectResult.getLeftTopPosition(); Point rightTop = detectResult.getRightTopPosition(); Point leftBottom = detectResult.getLeftBottomPosition(); Point rightBottom = detectResult.getRightBottomPosition(); List coordinates = new ArrayList(); coordinates.add(leftTop); coordinates.add(rightTop); coordinates.add(rightBottom); coordinates.add(leftBottom); // TODO 根据生成点 Point 集合 创建矫正 MLDocumentSkewCorrectionCoordinateInput 对象 MLDocumentSkewCorrectionCoordinateInput coordinateData = new MLDocumentSkewCorrectionCoordinateInput(coordinates); // TODO 根据 MLFrame 和 MLDocumentSkewCorrectionCoordinateInput 调用MLDocumentSkewCorrectionAnalyzer 的矫正接口 asyncDocumentSkewCorrect Task correctionTask = skewAnalyzer.asyncDocumentSkewCorrect(frame, coordinateData); // TODO 在回调接口监听onSuccess中 将矫正成功图片的Bitmap对象,使用 mIvSkewResult 展示矫正图片 correctionTask.addOnSuccessListener(new OnSuccessListener() { @Override public void onSuccess(MLDocumentSkewCorrectionResult refineResult) { mIvSkewResult.setImageBitmap(refineResult.getCorrected()); // 检测成功。 } }).addOnFailureListener(new OnFailureListener() { @Override public void onFailure(Exception e) { // 检测失败。 } }); } @Override protected void onActivityResult(int requestCode, int resultCode, Intent data) { super.onActivityResult(requestCode, resultCode, data); // 请不要删除以及修改,否则影响考试成绩 if (requestCode == REQUEST_IMG) { if (data != null && data.getExtras() != null) { Bundle bundle = data.getExtras(); if (bundle.containsKey("imageId")) { mImageId = bundle.getInt("imageId"); createImageBySelectedImageId(); } } } } } |
CopyRight 2018-2019 实验室设备网 版权所有 |