在 Node.js 中使用 OpenCV 加载图像列表

Loading a list of images with OpenCV in Node.js

本文关键字：OpenCV 加载 图像 列表 Node js      更新时间：2023-09-26

我正在尝试加载大量图像，这些图像以以下格式列在CVS文件中：

./path/to/img1.ext;label1
./path/to/img2.ext;label2

这是我写的脚本：

var cv = require("opencv"),
    fs = require("fs"),
    console = require("console"),
    util = require("util"),
    lazy = require("lazy.js");
var basePath = '/some/path/';
var csvFile = fs.createReadStream(basePath + 'db.csv', {flags:'r'});
var images = [],
    labels = [];
lazy(csvFile)
.lines()
.each(function(l) {
    var d = lazy(l).split(';').toArray();
    cv.readImage(basePath + d[0], function(e, m) {
        images.push(m);
    });
    labels.push(d[1]);
});
console.log(util.inspect(images));
console.log(util.inspect(labels));

它打印两行包含空数组[]的表示形式。

图像实际上是由OpenCV加载的，因为如果您在将m推入数组之前尝试打印它，它会正确打印[Matrix HxW ]，其中H和W代表图像的高度和宽度。

编辑：另外，您能想到比2个单独的数组更好的方法来保持每个图像与其标签相关联吗？

编辑：问题似乎是图像是异步加载的。所以问题是我缺乏异步编程的经验。我怎样才能做到这一点？

这里有一个使用管道和CSV2和THROUGH2库的节点解决方案，你可以在这里找到 https://github.com/rvagg/csv2，在这里 https://github.com/rvagg/through2

我使用 setTimeout 使用模拟异步函数对此进行了测试，它奏效了。 但是，因为我没有您的数据文件，所以我无法准确测试它。 如果有问题，请告诉我。

注意，我创建了一个对象数组。 每个对象都有图像及其标签。 我认为这是比尝试保留具有这些关联的两个数组更好的解决方案。 通常，如果您需要与数据建立关系，则一个对象将优于两个数组:)

var fs = require('fs');
var files = [];
var file = fs.createReadStream('test.txt');
var csv2 = require('csv2');
var th2 = require('through2');
var cv = require('opencv');
file
.pipe(csv2({'separator': ';'})).pipe(th2({objectMode: true},function(parsedLine, enc, callback){
    var me = this;
    cv.readImage(parsedLine[0], function(e, img) {
        files.push({image: img, label: parsedLine[1]});
        me.push(parsedLine);
        callback();
    });
}))
.on('data', function(data){/*do something with data if you want to*/})
.on('end', function(){console.log(files);});
相关文章：