LabeledFileList.h 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  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. /**
  17. * @brief reads images from directory
  18. */
  19. class LabeledFileList
  20. {
  21. private:
  22. bool debug_dataset;
  23. public:
  24. /** simple constructor */
  25. LabeledFileList();
  26. /** simple destructor */
  27. virtual ~LabeledFileList();
  28. LocalizationResult *getLocalizationInfo ( const ClassNames & classnames,
  29. int classno,
  30. const std::string & file,
  31. const NICE::Config & conf ) const;
  32. void get ( const std::string & dir,
  33. const NICE::Config & datasetconf,
  34. const ClassNames & classnames,
  35. LabeledSet & ls,
  36. bool localizationInfoDisabled = false,
  37. bool debugDataset = false );
  38. void getFromPattern ( const std::string & dir,
  39. const NICE::Config & datasetconf,
  40. const ClassNames & classnames,
  41. LabeledSet & ls,
  42. bool localizationInfoDisabled = false ) const;
  43. void getFromList ( const std::string & filelist,
  44. const NICE::Config & datasetconf,
  45. const ClassNames & classnames,
  46. LabeledSet & ls,
  47. bool localizationInfoDisabled = false ) const;
  48. };
  49. } // namespace
  50. #endif