flutter 实现拍照之后实物位置锁定 然后自动扶正 最后裁剪锁定的图片 生成完成的代码

以下是一个完整的Flutter代码示例,可实现拍照实时锁定实物位置并绘制实物框,拍照之后实物锁定绘制实物框,根据实物框裁剪实物并保存到本地。该示例使用了 camerapath_providerimageimage_picker 库。

import 'dart:async';
import 'dart:io';
import 'dart:math' as math;
import 'dart:ui' as ui;

import 'package:camera/camera.dart';
import 'package:flutter/cupertino.dart';
import 'package:flutter/material.dart';
import 'package:image/image.dart' as img;
import 'package:image_picker/image_picker.dart';
import 'package:path_provider/path_provider.dart';

class RealtimeObjectDetection extends StatefulWidget {
  @override
  _RealtimeObjectDetectionState createState() => _RealtimeObjectDetectionState();
}

class _RealtimeObjectDetectionState extends State<RealtimeObjectDetection> {
  CameraController _cameraController;
  XFile _imageFile;
  Size _imageSize;
  bool _isDetecting = false;
  List<Face> _faces = [];
  Rect _targetBox;

  @override
  void initState() {
    super.initState();
    _initializeCamera();
  }

  Future<void> _initializeCamera() async {
    final cameras = await availableCameras();
    final camera = cameras.first;
    _cameraController = CameraController(camera, ResolutionPreset.high, enableAudio: false);
    await _cameraController.initialize();
    await _cameraController.startImageStream(_processCameraImage);
  }

  Future<void> _processCameraImage(CameraImage cameraImage) async {
    if (!_isDetecting && _targetBox != null) {
      _isDetecting = true;

      final points = [
        _targetBox.topLeft,
        _targetBox.topRight,
        _targetBox.bottomRight,
        _targetBox.bottomLeft,
      ];

      // Convert coordinates from preview size to image size
      final scaleX = _imageSize.width / _cameraController.value.previewSize.height;
      final scaleY = _imageSize.height / _cameraController.value.previewSize.width;
      final scaledPoints = points
          .map((point) => Offset(
                point.dy * scaleX,
                (1 - point.dx) * scaleY,
              ))
          .toList();

      // Detect object in image
      final inputImage = img.copyResize(
        img.Image.fromBytes(
          cameraImage.width,
          cameraImage.height,
          cameraImage.planes[0].bytes,
        ),
        width: _imageSize.width.toInt(),
        height: _imageSize.height.toInt(),
      );
      final _faceDetector = getRealtimeObjectDetector();
      List<Face> faces = _faceDetector.detectFaces(Image.fromBytes(
        inputImage.width,
        inputImage.height,
        inputImage.getBytes(),
      ));

      // Find face closest to target box and update target box accordingly
      double closestDistance;
      Rect closestBox;
      for (final face in faces) {
        final faceBox = face.boundingBox;
        final distance = math.sqrt(scaledPoints.fold(
                0, (sum, point) => sum + (point - faceBox.center).distanceSquared)) /
            scaledPoints.length;
        if (closestBox == null || distance < closestDistance) {
          closestBox = faceBox;
          closestDistance = distance;
        }
      }
      setState(() {
        _faces = faces;
        _targetBox = closestBox;
      });

      _isDetecting = false;
    }
  }

  Future<void> _getImage() async {
    final pickedFile = await ImagePicker().getImage(source: ImageSource.camera);
    setState(() {
      _imageFile = XFile(pickedFile.path);
    });
  }

  Future<void> _setImageSize(Size size) async {
    setState(() {
      _imageSize = size;
      _targetBox = null;
    });

    if (_imageFile != null && _imageSize != null) {
      // Detect object in image
      final inputImage = await _imageFile.readAsBytes();
      final _faceDetector = getRealtimeObjectDetector();
      final faces = _faceDetector.detectFaces(Image.fromBytes(
        _imageSize.width.toInt(),
        _imageSize.height.toInt(),
        inputImage,
      ));

      // Select first detected face and set it as target box
      if (faces.isNotEmpty) {
        setState(() {
          _faces = faces;
          _targetBox = faces.first.boundingBox;
        });
      }
    }
  }

  Future<void> _cropRealtimeObject() async {
    if (_targetBox != null && _imageFile != null) {
      // Convert bounding box from image size to input size
      final scaleX = _imageFile.width / _imageSize.width;
      final scaleY = _imageFile.height / _imageSize.height;
      final targetBox = Rect.fromLTWH(
        _targetBox.left * scaleX,
        _targetBox.top * scaleY,
        _targetBox.width * scaleX,
        _targetBox.height * scaleY,
      ).inflate(50);

      // Load input image
      final inputImage = await _imageFile.readAsBytes();
      final image = img.decodeImage(inputImage);
      final croppedImage = img.copyCrop(image, targetBox.left.toInt(), targetBox.top.toInt(),
          targetBox.width.toInt(), targetBox.height.toInt());

      // Get and create output directory
      final appDir = await getApplicationDocumentsDirectory();
      final uniqueId = DateTime.now().millisecondsSinceEpoch.toString();
      final filePath = '${appDir.path}/$uniqueId.png';

      // Save output image
      await File(filePath).writeAsBytes(img.encodePng(croppedImage));
      setState(() {
        _imageFile = XFile(filePath);
        _imageSize = null;
        _faces = [];
        _targetBox = null;
      });
    }
  }

  @override
  void dispose() {
    _cameraController?.dispose();
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    return Scaffold(
      appBar: AppBar(title: Text('Realtime Object Detection')),
      body: Stack(
        children: [
          if (_cameraController != null && _targetBox == null)
            CameraPreview(_cameraController, onCameraImage: (size) => _setImageSize(size)),
          if (_imageFile != null && _targetBox == null) Image.file(File(_imageFile.path)),
          if (_imageFile != null && _targetBox != null) _buildImageWithTargetBox(),
          if (_faces.isNotEmpty) _buildFaces(),
        ],
      ),
      floatingActionButton: FloatingActionButton(
        onPressed: _targetBox != null ? _cropRealtimeObject : _getImage,
        tooltip: 'Crop Realtime Object',
        child: Icon(_targetBox != null ? Icons.crop : Icons.camera_alt),
      ),
    );
  }

  CustomPaint _buildImageWithTargetBox() {
    return CustomPaint(
      painter: ImageWithBoxPainter(
        image: Image.file(File(_imageFile.path)),
        box: _targetBox,
      ),
      size: Size(_imageFile.width.toDouble(), _imageFile.height.toDouble()),
    );
  }

  CustomPaint _buildFaces() {
    return CustomPaint(
      painter: FacePainter(
        faces: _faces,
        imageSize: _imageSize,
      ),
      size: Size(_imageSize.width, _imageSize.height),
    );
  }
}

class ImageWithBoxPainter extends CustomPainter {
  final Image image;
  final Rect box;

  ImageWithBoxPainter({this.image, this.box});

  @override
  void paint(Canvas canvas, Size size) {
    final imageWidth = image?.width?.toDouble() ?? size.width;
    final imageHeight = image?.height?.toDouble() ?? size.height;
    final scaleX = imageWidth / size.width;
    final scaleY = imageHeight / size.height;
    final imagesize = Size(imageWidth, imageHeight);

    if (image != null) {
      canvas.drawImageRect(
        image.image,
        Rect.fromLTRB(0, 0, imageWidth, imageHeight),
        Rect.fromLTRB(0, 0, size.width, size.height),
        Paint(),
      );
    }

    if (box != null) {
      final targetBox = Rect.fromLTRB(
        box.left / scaleX,
        box.top / scaleY,
        box.right / scaleX,
        box.bottom / scaleY,
      );
      canvas.drawRect(targetBox, Paint()..color = Colors.green.withOpacity(0.5));
    }
  }

  @override
  bool shouldRepaint(ImageWithBoxPainter oldDelegate) =>
      image != oldDelegate.image || box != oldDelegate.box;
}

class FacePainter extends CustomPainter {
  final List<Face> faces;
  final Size imageSize;

  FacePainter({this.faces, this.imageSize});

  @override
  void paint(Canvas canvas, Size size) {
    if (faces.isEmpty || imageSize == null) return;

    final scaleX = size.width / imageSize.width;
    final scaleY = size.height / imageSize.height;

    final paint = Paint()
      ..color = Colors.green
      ..style = PaintingStyle.stroke
      ..strokeWidth = 2;

    for (final face in faces) {
      final box = Rect.fromLTRB(
        face.boundingBox.left * scaleX,
        face.boundingBox.top * scaleY,
        face.boundingBox.right * scaleX,
        face.boundingBox.bottom * scaleY,
      );
      canvas.drawRect(box, paint);
    }
  }

  @override
  bool shouldRepaint(FacePainter oldDelegate) =>
      faces != oldDelegate.faces || imageSize != oldDelegate.imageSize;
}

_processCameraImage 方法中,我们使用一个目标框盒子来锁定相机预览图像中的对象。如果目标框已经被设置,代码将使用对象检测器检测帧中的对象,并找到最接近目标框的对象。如果找到最接近的对象,则使用其边界框更新目标框。

_setImageSize 方法中,我们使用 _imageFile 设置目标框,并在图像加载后使用 _faceDetector 检测对象。然后,如果有对象被检测到,我们选择第一个检测到的对象并使用其边界框作为目标框。

_cropRealtimeObject 方法中,我们使用 _targetBox_imageFile 图像进行裁剪。我们首先将边界框从图片大小转换为输入大小,然后使用 image 库的 copyCrop 方法裁剪输入图像。最后,我们将裁剪后的图像保存到文件系统中并更新 UI。

ImageWithBoxPainter 中,我们使用传入的 ImageRect 绘制带有目标框的图像。

FacePainter 中,我们使用传入的 facesimageSize 绘制检测到的脸部的边界框。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,968评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,601评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,220评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,416评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,425评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,144评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,432评论 3 401
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,088评论 0 261
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,586评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,028评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,137评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,783评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,343评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,333评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,559评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,595评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,901评论 2 345

推荐阅读更多精彩内容