LabeledFileList.h 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. /**
  2. * @file LabeledFileList.h
  3. * @brief reads images from directory
  4. * @author Erik Rodner
  5. * @date 17.09.2007
  6. */
  7. #ifndef LabeledFileListINCLUDE
  8. #define LabeledFileListINCLUDE
  9. #include <string>
  10. #include "core/basics/Config.h"
  11. #include "vislearning/baselib/ProgressBar.h"
  12. #include "ClassNames.h"
  13. #include "LocalizationResult.h"
  14. #include "LabeledSet.h"
  15. namespace OBJREC {
  16. /** reads images from directory */
  17. class LabeledFileList
  18. {
  19. private:
  20. bool debug_dataset;
  21. public:
  22. /** simple constructor */
  23. LabeledFileList();
  24. /** simple destructor */
  25. virtual ~LabeledFileList();
  26. LocalizationResult *getLocalizationInfo ( const ClassNames & classnames,
  27. int classno,
  28. const std::string & file,
  29. const NICE::Config & conf ) const;
  30. void get ( const std::string & dir,
  31. const NICE::Config & datasetconf,
  32. const ClassNames & classnames,
  33. LabeledSet & ls,
  34. bool localizationInfoDisabled = false,
  35. bool debugDataset = false );
  36. void getFromPattern ( const std::string & dir,
  37. const NICE::Config & datasetconf,
  38. const ClassNames & classnames,
  39. LabeledSet & ls,
  40. bool localizationInfoDisabled = false ) const;
  41. void getFromList ( const std::string & filelist,
  42. const NICE::Config & datasetconf,
  43. const ClassNames & classnames,
  44. LabeledSet & ls,
  45. bool localizationInfoDisabled = false ) const;
  46. };
  47. } // namespace
  48. #endif