Swin2sr Compressed Sr X4 48

🚀 画像拡大モデル (Swin2SR)

Transformers.jsと互換性のあるONNX形式の画像拡大モデルです。

🚀 クイックスタート

このモデルは、caidas/swin2SR-compressed-sr-x4-48 のONNXウェイトを使用し、Transformers.jsと互換性を持たせています。

📦 インストール

まだの場合は、Transformers.js JavaScriptライブラリを NPM から以下のコマンドでインストールできます。

npm i @xenova/transformers

💻 使用例

基本的な使用法

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

// 画像から画像へのパイプラインを作成
const upscaler = await pipeline('image-to-image', 'Xenova/swin2SR-compressed-sr-x4-48', {
    // quantized: false, // この行のコメントを外すと、量子化バージョンを使用できます
});

// 画像を拡大
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/butterfly.jpg';
const output = await upscaler(url);
// RawImage {
//   data: Uint8Array(3145728) [ ... ],
//   width: 1024,
//   height: 1024,
//   channels: 3
// }

// (オプション) 拡大した画像を保存
output.save('upscaled.png');

出力例

📚 詳細ドキュメント

Note: WebMLが普及するまで、ONNXウェイトを別のリポジトリにするのは一時的な解決策です。モデルをWebで使用可能にする場合は、🤗 Optimum を使用してONNXに変換し、このリポジトリのように構成することをおすすめします (ONNXウェイトは onnx というサブフォルダに配置します)。

📄 情報テーブル