swh:1:snp:67458089460e0f4324e7769efad93379abfcf654
Tip revision: 8cf122b9c7f7403efe90dca3b0af0ebfec10b26d authored by jzhzhang on 07 January 2024, 04:10:10 UTC
Update pose_estimation.cpp
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