swh:1:snp:67458089460e0f4324e7769efad93379abfcf654
Raw File
Tip revision: 8cf122b9c7f7403efe90dca3b0af0ebfec10b26d authored by jzhzhang on 07 January 2024, 04:10:10 UTC
Update pose_estimation.cpp
Tip revision: 8cf122b
DataReader.h
#ifndef DATAREADER_H
#define DATAREADER_H


#include <iostream>
#include <stdio.h>
#include <string>
#include <opencv2/opencv.hpp>



class DataReader
{
    public:
    DataReader(std::string file, bool flipColors);
    virtual ~DataReader();
    int getFramesNum();
    void getNextFrame(cv::Mat& Color_mat,cv::Mat& Depth_mat);
    bool hasMore();

    private:
    std::vector<cv::Mat> v_color;
    std::vector<cv::Mat> v_depth;

    void readFile();
    FILE * fp;
    int numFrames;
    int height;
    int width;
    int currentFrame;

};

#endif
back to top