mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
fix: improve defringe quality and add defensive guards in transparency-fixer
This commit is contained in:
@@ -50,7 +50,7 @@ async function applyDefringe(buffer: Buffer, intensity: number): Promise<Buffer>
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Blur the alpha channel
|
// Blur the alpha channel
|
||||||
const blurRadius = Math.max(1, Math.round(intensity / 20));
|
const blurRadius = Math.max(0.3, Math.round(intensity / 20));
|
||||||
const blurredAlphaRaw = await sharp(alpha, {
|
const blurredAlphaRaw = await sharp(alpha, {
|
||||||
raw: { width: info.width, height: info.height, channels: 1 },
|
raw: { width: info.width, height: info.height, channels: 1 },
|
||||||
})
|
})
|
||||||
@@ -63,6 +63,9 @@ async function applyDefringe(buffer: Buffer, intensity: number): Promise<Buffer>
|
|||||||
const result = Buffer.from(data);
|
const result = Buffer.from(data);
|
||||||
for (let i = 0; i < pixelCount; i++) {
|
for (let i = 0; i < pixelCount; i++) {
|
||||||
if (alpha[i] > 0 && blurredAlphaRaw[i] < threshold) {
|
if (alpha[i] > 0 && blurredAlphaRaw[i] < threshold) {
|
||||||
|
result[i * 4] = 0;
|
||||||
|
result[i * 4 + 1] = 0;
|
||||||
|
result[i * 4 + 2] = 0;
|
||||||
result[i * 4 + 3] = 0;
|
result[i * 4 + 3] = 0;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -96,6 +99,9 @@ async function processTransparencyFix(
|
|||||||
const isOom = err instanceof Error && err.message.includes("out of memory");
|
const isOom = err instanceof Error && err.message.includes("out of memory");
|
||||||
if (!isOom) throw err;
|
if (!isOom) throw err;
|
||||||
|
|
||||||
|
// removeBackground has its own internal u2net fallback on OOM.
|
||||||
|
// This route-level fallback provides an intermediate quality step
|
||||||
|
// (birefnet-general) before that kicks in on a second OOM.
|
||||||
onProgress?.(5, `Retrying with fallback model (${FALLBACK_MODEL})`);
|
onProgress?.(5, `Retrying with fallback model (${FALLBACK_MODEL})`);
|
||||||
resultBuffer = await removeBackground(
|
resultBuffer = await removeBackground(
|
||||||
inputBuffer,
|
inputBuffer,
|
||||||
|
|||||||
Reference in New Issue
Block a user