1

我需要以像素比例编辑 34 000 张照片，所以我编写了一个程序，如果图像中的像素超过特定的 RBG 值（240、240、240），则替换它们，基本上我只想隔离白色像素并设置所有内容否则为黑色。这是我的代码：

const jimp = require('jimp')
var fs = require('fs');
const PNG = require('pngjs').PNG;


const frames = fs.readdirSync("./out/")
for (let i = 0; i < frames.length; i++) {
    function work(frames, i) {
        if (frames[i] !== ".DS_Store") { 
            jimp.read('./out/' + frames[i], (err, image) => {
                const bitmap = new Uint8ClampedArray(image.bitmap.data)
                let texterHighlit = 240
                let nextD = 4;
                let items = [];
                for (i = 0; i < bitmap.length + 1; i++) {
                    if (i === nextD) {
                        if (bitmap[i - 4] > texterHighlit && bitmap[i - 3] > texterHighlit && bitmap[i - 2] > texterHighlit) {
                            items.push(255)
                            items.push(255)
                            items.push(255)
                            items.push(255)
                        } else {
                            items.push(0)
                            items.push(0)
                            items.push(0)
                            items.push(255)
                        }
                        nextD = nextD + 4;
                    }
                }

                var img_width = image.bitmap.width;
                var img_height = image.bitmap.height;
                var img_data = Uint8ClampedArray.from(items);
                var img_png = new PNG({ width: img_width, height: img_height })
                img_png.data = Buffer.from(img_data);
                img_png.pack().pipe(fs.createWriteStream("./out/" + frames[i])) 
                return "done";

            })
        }
    }
    work(frames, i)

}

现在，这对一些图像非常有效，但是在我提到的 34 000 个图像文件夹等大型数据集上使用它时问题开始出现。

作为参考，我在带有 PM2 和--no-autorestart标志的 CentOS8 上运行此代码。当不使用 PM2 而只是使用node <file>时，进程被杀死并在控制台中输出“被杀死”，它不会显示“内存不足”。我已经尝试调试了很长一段时间，但没有办法阻止这种情况，我尝试不使用 Jimp，而是使用一个名为image-pixels的包，它产生了相同的结果。

有没有办法防止这种情况发生，或者我必须在较小的文件夹（例如每个 1000 个图像）上手动执行它？

4

1 回答 1

0

好吧，最终正确地完成了异步操作，这归功于NeonSurvivor#2046在 Discord 上给我的这个答案！

const jimp = require('jimp')
var fs = require('fs');

(async() => {
    async function work(frame) {
    if (frame === '.DS_Store') return;
    const image = await jimp.read('./out/' + frame);
    const texterHighlit = 240;
    image.scan(0, 0, image.bitmap.width, image.bitmap.height, (x, y, idx) => {
        const { data } = image.bitmap;
        data[idx] = data[idx + 1] = data[idx + 2] = 255 *  [0, 1, 2].every(offset => data[idx + offset] > texterHighlit);
        data[idx + 3] = 255;
    });
    image.write('./out/' + frame);
    return 'done';
}

for (const frame of fs.readdirSync('./out/')) {
    await work(frame);
    await new Promise(res => setTimeout(res, 500));
}
})()
于 2021-08-07T15:21:48.130 回答