81aefed13a
The format of the training set can be changed with the `-pngoutput` key. Output image will be resized to a 640x480 size if greater.
35 lines
730 B
C++
35 lines
730 B
C++
#ifndef IOUTPUT_H
|
|
#define IOUTPUT_H
|
|
|
|
#include <cstdio>
|
|
|
|
#include "_cvcommon.h"
|
|
|
|
struct CvMat;
|
|
struct CvRect;
|
|
|
|
class IOutput
|
|
{
|
|
public:
|
|
enum OutputType {PNG_TRAINING_SET, JPG_TEST_SET};
|
|
public:
|
|
virtual bool write( const CvMat& img,
|
|
const CvRect& boundingBox ) =0;
|
|
|
|
virtual ~IOutput();
|
|
|
|
static IOutput* createOutput( const char *filename, OutputType type );
|
|
protected:
|
|
IOutput();
|
|
/* finds the beginning of the last token in the path */
|
|
void findFilePathPart( char **partOfPath, char *fullPath );
|
|
virtual bool init( const char* filename );
|
|
protected:
|
|
int currentIdx;
|
|
char imgFullPath[PATH_MAX];
|
|
char* imgFileName;
|
|
FILE* annotationsList;
|
|
};
|
|
|
|
#endif // IOUTPUT_H
|