51 lines
1.2 KiB
C
51 lines
1.2 KiB
C
|
#ifndef _OPENCV_IMAGESTORAGE_H_
|
||
|
|
#define _OPENCV_IMAGESTORAGE_H_
|
||
|
|
|
||
|
|
|
||
|
|
class CvCascadeImageReader
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
bool create( const std::string _posFilename, const std::string _negFilename, cv::Size _winSize );
|
||
|
|
void restart() { posReader.restart(); }
|
||
|
|
bool getNeg(cv::Mat &_img) { return negReader.get( _img ); }
|
||
|
|
bool getPos(cv::Mat &_img) { return posReader.get( _img ); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
class PosReader
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
PosReader();
|
||
|
|
virtual ~PosReader();
|
||
|
|
bool create( const std::string _filename );
|
||
|
|
bool get( cv::Mat &_img );
|
||
|
|
void restart();
|
||
|
|
|
||
|
|
short* vec;
|
||
|
|
FILE* file;
|
||
|
|
int count;
|
||
|
|
int vecSize;
|
||
|
|
int last;
|
||
|
|
int base;
|
||
|
|
} posReader;
|
||
|
|
|
||
|
|
class NegReader
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
NegReader();
|
||
|
|
bool create( const std::string _filename, cv::Size _winSize );
|
||
|
|
bool get( cv::Mat& _img );
|
||
|
|
bool nextImg();
|
||
|
|
|
||
|
|
cv::Mat src, img;
|
||
|
|
std::vector<std::string> imgFilenames;
|
||
|
|
cv::Point offset, point;
|
||
|
|
float scale;
|
||
|
|
float scaleFactor;
|
||
|
|
float stepFactor;
|
||
|
|
size_t last, round;
|
||
|
|
cv::Size winSize;
|
||
|
|
} negReader;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|