当前位置：首页 > article >正文

鸿蒙人脸识别

article 2024/12/24 16:41:42

鸿蒙人脸识别

参考文档

鸿蒙原生提供用于实现人脸识别的API。感觉能够实现一些好玩的效果。

请添加图片描述

private detectFace(img: PixelMap) {
  // 调用人脸检测接口
  faceDetector.init();
  let visionInfo: faceDetector.VisionInfo = {
    pixelMap: img,
  };
  faceDetector.detect(visionInfo)
    .then((data: faceDetector.Face[]) => {
      if (data.length === 0) {
      } else {
        let faceString = JSON.stringify(data);
        hilog.info(0x0000, 'faceDetectorSample', "faceString data is " + faceString);
        promptAction.showDialog({
          message: `识别到信息  ${faceString}`
        })
      }
    })
    .catch((error: BusinessError) => {
      hilog.error(0x0000, 'faceDetectorSample', `Face detection failed. Code: ${error.code}, message: ${error.message}`);
    });
  faceDetector.release();
}

示例代码

import { BusinessError } from '@kit.BasicServicesKit';
import { hilog } from '@kit.PerformanceAnalysisKit';
import { photoAccessHelper } from '@kit.MediaLibraryKit';
import { image } from '@kit.ImageKit';
import { fileIo } from '@kit.CoreFileKit';
import { faceDetector } from '@kit.CoreVisionKit';
import { promptAction } from '@kit.ArkUI';

const TAG: string = "index";

@Entry
@Component
struct Index {
  @State chooseImage: image.PixelMap | null = null
  @State segImage: image.PixelMap | null = null

  private openPhoto(): Promise<Array<string>> {
    return new Promise<Array<string>>((resolve, reject) => {
      let PhotoSelectOptions = new photoAccessHelper.PhotoSelectOptions();
      PhotoSelectOptions.MIMEType = photoAccessHelper.PhotoViewMIMETypes.IMAGE_TYPE;
      PhotoSelectOptions.maxSelectNumber = 1;
      let photoPicker: photoAccessHelper.PhotoViewPicker = new photoAccessHelper.PhotoViewPicker();
      hilog.info(0x0000, TAG, 'PhotoViewPicker.select successfully, PhotoSelectResult uri: ');
      photoPicker.select(PhotoSelectOptions).then((PhotoSelectResult) => {
        hilog.info(0x0000, TAG, `PhotoViewPicker.select successfully, PhotoSelectResult uri: ${PhotoSelectResult.photoUris}`);
        resolve(PhotoSelectResult.photoUris)
      }).catch((err: BusinessError) => {
        hilog.error(0x0000, TAG, `PhotoViewPicker.select failed with errCode: ${err.code}, errMessage: ${err.message}`);
        reject();
      });
    })
  }

  private async loadImage(path: string) {
    let imageSource: image.ImageSource | undefined = undefined
    let fileSource = await fileIo.open(path, fileIo.OpenMode.READ_ONLY)
    imageSource = image.createImageSource(fileSource.fd)
    this.chooseImage = await imageSource.createPixelMap()
    hilog.info(0x0000, TAG, `this.chooseImage===${this.chooseImage}`);
  }

  private detectFace(img: PixelMap) {
    // 调用人脸检测接口
    faceDetector.init();
    let visionInfo: faceDetector.VisionInfo = {
      pixelMap: img,
    };
    faceDetector.detect(visionInfo)
      .then((data: faceDetector.Face[]) => {
        if (data.length === 0) {
        } else {
          let faceString = JSON.stringify(data);
          hilog.info(0x0000, 'faceDetectorSample', "faceString data is " + faceString);
          promptAction.showDialog({
            message: `识别到信息  ${faceString}`
          })
        }
      })
      .catch((error: BusinessError) => {
        hilog.error(0x0000, 'faceDetectorSample', `Face detection failed. Code: ${error.code}, message: ${error.message}`);
      });
    faceDetector.release();
  }

  build() {
    Column() {
      Image(this.chooseImage)
        .objectFit(ImageFit.Contain)
        .width('100%')
        .height('40%')

      Row() {
        Image(this.segImage)
          .objectFit(ImageFit.Contain)
          .width('100%')
          .height('100%')
      }
      .width('100%')
      .height('40%')

      Blank()

      Row() {
        Text('选择图片')
          .height(50)
          .width(140)
          .borderRadius(25)
          .textAlign(TextAlign.Center)
          .backgroundColor(Color.Orange)
          .onClick(async () => {
            let urls = await this.openPhoto()
            if (urls.length > 0) {
              let url = urls[0]
              this.loadImage(url)
            }
          })

        Text('人脸识别')
          .height(50)
          .width(140)
          .borderRadius(25)
          .textAlign(TextAlign.Center)
          .backgroundColor(Color.Orange)
          .onClick(async () => {
            if (this.chooseImage) {
              this.detectFace(this.chooseImage)
            }
          })
      }
    }
    .height('100%')
    .width('100%')
  }
}