Update MAXIM models to use patch sizes (#1278)

thekevinscott · Dec 1, 2023 · 996ceae · 996ceae
1 parent d456542
commit 996ceae
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 8 deletions.
diff --git a/models/maxim-deblurring/demo/index.js b/models/maxim-deblurring/demo/index.js
@@ -1,12 +1,12 @@
 import Upscaler from "upscaler";
 import model from '@upscalerjs/maxim-deblurring';
-import fixture from "./fixture.png";
 
 const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+const fixture = document.getElementById('fixture');
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);

diff --git a/models/maxim-dehazing-indoor/demo/index.js b/models/maxim-dehazing-indoor/demo/index.js
@@ -6,7 +6,7 @@ const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);

diff --git a/models/maxim-dehazing-outdoor/demo/index.js b/models/maxim-dehazing-outdoor/demo/index.js
@@ -6,7 +6,7 @@ const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);

diff --git a/models/maxim-denoising/demo/index.js b/models/maxim-denoising/demo/index.js
@@ -6,7 +6,7 @@ const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);

diff --git a/models/maxim-deraining/demo/index.js b/models/maxim-deraining/demo/index.js
@@ -6,7 +6,7 @@ const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);

diff --git a/models/maxim-enhancement/demo/index.js b/models/maxim-enhancement/demo/index.js
@@ -6,7 +6,7 @@ const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);

diff --git a/models/maxim-retouching/demo/index.js b/models/maxim-retouching/demo/index.js
@@ -6,7 +6,7 @@ const upscaler = new Upscaler({
   model,
 });
 
-upscaler.upscale(fixture).then((upscaledImgSrc) => {
+upscaler.upscale(fixture, { patchSize: 64, padding: 2, progress: console.log }).then((upscaledImgSrc) => {
   const img = document.createElement("img");
   img.src = upscaledImgSrc;
   document.getElementById("target").appendChild(img);