lib-colored-image 🎨

Uma biblioteca leve em JavaScript para colorir imagens antigas em preto e branco utilizando IA (ONNX Runtime) e aceleração de hardware (WebGPU/WebGL). Lightweight JavaScript library for colorizing old black and white images using AI (ONNX Runtime).

✨ Características

• Leve: Sem dependências pesadas como o TensorFlow.js completo.
• Rápida: Utiliza o motor ONNX Runtime Web.
• Precisa: Processamento no espaço de cor LAB para melhores resultados.

🚀 Instalação

Instale via npm Install with npm: npm install lib-colored-image

Como usar

Primeiro, deve ter um modelo .onnx disponível. First, you must have an .onnx template available... https://github.com/onnx/models

JavaScript

import { Colorizer } from 'lib-colored-image';

const colorizer = new Colorizer();

// Inicializa o modelo // Start the model await colorizer.loadModel('/path/to/your/model.onnx');

// Colore uma imagem do DOM // Color an image from the DOM. const image = document.getElementById('minha-foto'); const result = await colorizer.colorize(image);

// 'result' pode ser desenhado num canvas ou processado // the result can be drawn on a canvas or processed console.log('Imagem colorida com sucesso!');

🛠️ API

Carregar o arquivo do modelo pré-treinado. Load the pre-trained model file: loadModel(url)

Receber um elemento de imagem e retorna os dados coloridos (canais AB previstos). Receives an image element and returns the color data (AB channels): colorize(htmlImageElement)

Licença

MIT