HDFql 保存和加载图像

这是基于@SOG给出的答案 - 谢谢！

#include <iostream.h>
#include <HDFql.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
int main (int argc, const char * argv[]) {
char script[1024];
//Create HDF5 file and group "test"
HDFql::execute("CREATE TRUNCATE FILE /tmp/test.h5");
HDFql::execute("USE FILE /tmp/test.h5");
HDFql::execute("CREATE GROUP test");
//==== Load Image, save it to test.h5/test/image ====
//Choose color or grayscale
//cv::ImreadModes read_mode = cv::IMREAD_GRAYSCALE;
cv::ImreadModes read_mode = cv::IMREAD_COLOR;
cv::Mat img = cv::imread("/tmp/lena.jpg", read_mode);
cv::Size size(img.size());
int num_pixels = size.width*size.height*img.channels();
cv::imshow("image", img);
cv::waitKey();
sprintf(script, "CREATE DATASET test/image AS UNSIGNED TINYINT(%d) "
"VALUES FROM MEMORY %d", num_pixels,
HDFql::variableTransientRegister(img.data));
HDFql::execute(script);
HDFql::execute("CLOSE FILE");
//==== Load HDF5, and load saved image data to a new cv::Mat image ====
HDFql::execute("USE FILE /tmp/test.h5");
cv::Mat img_loaded;
if(read_mode == cv::IMREAD_COLOR)
{
img_loaded = cv::Mat::zeros(size, CV_8UC3);
} else
{
img_loaded = cv::Mat::zeros(size, CV_8UC1);
}
std::cout << "size = " << img_loaded.size() << ", type = " << img_loaded.type() 
<< ", channels = " << img_loaded.channels() <<std::endl;
sprintf(script, "SELECT FROM test/image INTO MEMORY %d",
HDFql::variableTransientRegister(img_loaded.data));
HDFql::execute(script);
HDFql::execute("CLOSE FILE");
cv::imshow("loaded image", img_loaded);
cv::waitKey();
}

此代码非常适合灰度和彩色图像。@SOG的答案仍然存在一些问题，除了一些小事情，我无法让它与数据类型OPAQUE一起使用，所以我又回到了使用UNSIGNED TINYINT.我不知道这是为什么。

除此之外，@SOG的彻底回答几乎抓住了我的代码不起作用的原因。

代码片段中的 HDFqlSELECT操作包含两个问题：1( 无法指定AS INT,2( 数据集image位于组test中。此外，永远不会调用(即执行(将数据写入数据集的脚本。

此外，还可以对代码片段进行一些优化，即：

1. 删除关键字"CONTIGUOUS"，因为在创建数据集时它是 默认情况下是连续的。
2. 将数据集的数据类型更改为OPAQUE- 我相信您希望存储图像的原始数据，因此此数据类型更合适。
3. 以瞬态方式注册变量，以便在使用它们(由 HDFql(后自动取消注册 - 这将减轻您执行此操作的负担。

以下是更正/重构的代码片段：

#include <iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <HDFql.hpp>
int main() {
char script[1024];
HDFql::execute("CREATE AND USE FILE /tmp/test.h5");
HDFql::execute("CREATE GROUP image");
cv::Mat img = cv::imread("/tmp/lena.jpg", CV_8UC3);
cv::Size size(img.size());
std::cout << "Size = " << size.width << ", " << size.height;
for(int i=0; i<100; i++)
{
// Show a few pixel values - cast is necessary, since values are unsigned char
std::cout << i << ": " << (int)img.data[i] << std::endl;
}
sprintf(script, "CREATE DATASET test/image AS OPAQUE(%d) VALUES FROM MEMORY %d", size.width*size.height*3, HDFql::variableTransientRegister(img.data));
HDFql::execute(script);
HDFql::execute("CLOSE FILE");
cv::Mat img_loaded = cv::Mat::zeros(cv::Size(size.width, size.height), CV_8UC3);
sprintf(script, "SELECT FROM /tmp/test.h5 test/image INTO MEMORY %d", HDFql::variableTransientRegister(img_loaded.data));
HDFql::execute(script);
cv::imshow("loaded image", img_loaded);
cv::waitKey();
return 0;
}