Swin2sr Lightweight X2 64

🚀 画像拡大モデル

このプロジェクトは、caidas/swin2SR-lightweight-x2-64 ベースモデルを使用し、Transformers.js ライブラリと互換性のある ONNX ウェイトを提供します。画像の拡大処理を行うことができます。

🚀 クイックスタート

このリポジトリは、caidas/swin2SR-lightweight-x2-64 の ONNX ウェイトを使用しており、Transformers.js と互換性があります。

📦 インストール

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

npm i @xenova/transformers

💻 使用例

基本的な使用法

Xenova/swin2SR-lightweight-x2-64 を使用して画像を拡大する例です。

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

// Create image-to-image pipeline
const upscaler = await pipeline('image-to-image', 'Xenova/swin2SR-lightweight-x2-64', {
    // quantized: false, // Uncomment this line to use the quantized version
});

// Upscale an image
const url = 'https://huggingface.co/datasets/Xenova/transformers.js-docs/resolve/main/butterfly.jpg';
const output = await upscaler(url);
// RawImage {
//   data: Uint8Array(786432) [ ... ],
//   width: 512,
//   height: 512,
//   channels: 3
// }

// (Optional) Save the upscaled image
output.save('upscaled.png');

⚠️ 重要提示

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