许可协议: gpl-3.0
库名称: transformers.js
标签:

• apisr
• 超分辨率
  任务类型: 图像到图像

https://github.com/Kiteretsu77/APISR 附带ONNX权重，以兼容Transformers.js。

使用方法 (Transformers.js)

如果尚未安装，可以通过以下命令从NPM安装Transformers.js JavaScript库：

npm i @xenova/transformers

示例: 使用Xenova/4x_APISR_GRL_GAN_generator-onnx放大图像。

import { pipeline } from '@xenova/transformers';

// 创建图像到图像的处理管道
const upscaler = await pipeline('image-to-image', 'Xenova/4x_APISR_GRL_GAN_generator-onnx', {
    quantized: false,
});

// 放大图像
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/anime.png';
const output = await upscaler(url);
// RawImage {
//   data: Uint8Array(16588800) [ ... ],
//   width: 2560,
//   height: 1920,
//   channels: 3
// }

// (可选) 保存放大后的图像
output.save('upscaled.png');

注：为ONNX权重单独设立仓库是临时解决方案，直到WebML获得更多支持。如果您希望使您的模型适用于Web，建议使用🤗 Optimum转换为ONNX格式，并按照此仓库的结构（ONNX权重位于名为onnx的子文件夹中）组织您的仓库。