Flutter超分辨率处理插件flutter_super_resolution的使用

简介

flutter_super_resolution 是一个用于使用ONNX Runtime神经网络进行图像放大处理的Flutter库。它支持高级平铺处理。

平台支持

• Windows
• macOS
• iOS
• Android（仅限ARM设备）

特性

• 使用机器学习模型进行高质量的图像放大
• 支持全图和平铺处理
• 可自定义的平铺大小和重叠
• 在放大过程中跟踪进度
• 针对移动设备进行了优化

安装

在 pubspec.yaml 文件中添加依赖项：

dependencies:
  flutter_super_resolution: ^1.0.0
  onnxruntime: ^latest_version

使用

以下是一个完整的示例代码，展示了如何使用 flutter_super_resolution 插件来放大图像：

import 'package:flutter/material.dart';
import 'package:flutter/services.dart';
import 'dart:ui' as ui;
import 'package:flutter_super_resolution/flutter_super_resolution.dart';

void main() {
  runApp(SuperResolutionDemoApp());
}

class SuperResolutionDemoApp extends StatefulWidget {
  [@override](/user/override)
  _SuperResolutionDemoAppState createState() => _SuperResolutionDemoAppState();
}

class _SuperResolutionDemoAppState extends State<SuperResolutionDemoApp> {
  ui.Image? _originalImage;
  ui.Image? _upscaledImage;
  double _progress = 0.0;
  String _progressMessage = '';
  bool _isProcessing = false;

  [@override](/user/override)
  void initState() {
    super.initState();
    _loadDefaultImage();
  }

  Future<void> _loadDefaultImage() async {
    final ByteData data = await rootBundle.load('assets/sample_image.jpg');
    final Uint8List bytes = data.buffer.asUint8List();
    final ui.Codec codec = await ui.instantiateImageCodec(bytes);
    final ui.FrameInfo fi = await codec.getNextFrame();

    setState(() {
      _originalImage = fi.image;
    });
  }

  Future<void> _upscaleImage() async {
    if (_originalImage == null) return;

    setState(() {
      _isProcessing = true;
      _progress = 0.0;
      _progressMessage = '';
    });

    final upscaler = FlutterUpscaler(
      tileSize: 128,
      overlap: 8,
    );

    try {
      // 初始化模型从资源文件
      await upscaler.initializeModel('assets/super_resolution_model.onnx');

      // 放大图像并追踪进度
      final upscaledImage = await upscaler.upscaleImage(
        _originalImage!,
        scale: 2,
        onProgress: (progress, message) {
          setState(() {
            _progress = progress;
            _progressMessage = message;
          });
        },
      );

      setState(() {
        _upscaledImage = upscaledImage;
        _isProcessing = false;
      });

      upscaler.dispose();
    } catch (e) {
      setState(() {
        _isProcessing = false;
      });
      print('放大失败: $e');
    }
  }

  [@override](/user/override)
  Widget build(BuildContext context) {
    return MaterialApp(
      home: Scaffold(
        appBar: AppBar(
          title: Text('超分辨率演示'),
        ),
        body: Column(
          mainAxisAlignment: MainAxisAlignment.center,
          children: [
            if (_originalImage != null)
              Row(
                mainAxisAlignment: MainAxisAlignment.spaceEvenly,
                children: [
                  Column(
                    children: [
                      Text('原始图像'),
                      Image(image: ui.ImageForRenderObject(image: _originalImage!)),
                    ],
                  ),
                  if (_upscaledImage != null)
                    Column(
                      children: [
                        Text('放大图像'),
                        Image(image: ui.ImageForRenderObject(image: _upscaledImage!)),
                      ],
                    ),
                ],
              ),
            if (_isProcessing)
              Column(
                children: [
                  LinearProgressIndicator(value: _progress),
                  Text(_progressMessage),
                ],
              ),
            ElevatedButton(
              onPressed: _isProcessing ? null : _upscaleImage,
              child: Text('放大图像'),
            ),
          ],
        ),
      ),
    );
  }
}