c++ mnist转化为opecv Mat

FishBear_move_on

于 2018-09-20 20:09:48 发布

阅读量1k

点赞数 1

文章标签： mnist数据集 mnist数据集转化为opencv mat mnist数据集显示

本文链接：https://blog.youkuaiyun.com/haluoluo211/article/details/82792814

版权

本文主要介绍如何使用C++将mnist 数据集转化为Opencv Mat，问题来源主要代码以及运行示例如下：

uint32_t swap_endian(uint32_t val) {
    val = ((val << 8) & 0xFF00FF00) | ((val >> 8) & 0xFF00FF);
    return (val << 16) | (val >> 16);
}

void read_mnist_cv(const char* image_filename, const char* label_filename){
    // Open files
    std::ifstream image_file(image_filename, std::ios::in | std::ios::binary);
    std::ifstream label_file(label_filename, std::ios::in | std::ios::binary);

    // Read the magic and the meta data
    uint32_t magic;
    uint32_t num_items;
    uint32_t num_labels;
    uint32_t rows;
    uint32_t cols;

    image_file.read(reinterpret_cast<char*>(&magic), 4);
    magic = swap_endian(magic);
    if(magic != 2051){
        cout<<"Incorrect image file magic: "<<magic<<endl;
        return;
    }

    label_file.read(reinterpret_cast<char*>(&magic), 4);
    magic = swap_endian(magic);
    if(magic != 2049){
        cout<<"Incorrect image file magic: "<<magic<<endl;
        return;
    }

    imag