HOCRCnn.h 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
  1. /***********************************************************
  2. * File generated by the HALCON-Compiler hcomp version 20.11
  3. * Usage: Interface to C++
  4. *
  5. * Software by: MVTec Software GmbH, www.mvtec.com
  6. ***********************************************************/
  7. #ifndef HCPP_HOCRCNN
  8. #define HCPP_HOCRCNN
  9. namespace HalconCpp
  10. {
  11. // Represents an instance of a CNN OCR classifier.
  12. class LIntExport HOCRCnn : public HHandle
  13. {
  14. public:
  15. // Create an uninitialized instance
  16. HOCRCnn():HHandle() {}
  17. // Copy constructor
  18. HOCRCnn(const HOCRCnn& source) : HHandle(source) {}
  19. // Copy constructor
  20. HOCRCnn(const HHandle& handle);
  21. // Create HOCRCnn from handle, taking ownership
  22. explicit HOCRCnn(Hlong handle);
  23. bool operator==(const HHandle& obj) const
  24. {
  25. return HHandleBase::operator==(obj);
  26. }
  27. bool operator!=(const HHandle& obj) const
  28. {
  29. return HHandleBase::operator!=(obj);
  30. }
  31. protected:
  32. // Verify matching semantic type ('ocr_cnn')!
  33. virtual void AssertType(Hphandle handle) const;
  34. public:
  35. // Deep copy of all data represented by this object instance
  36. HOCRCnn Clone() const;
  37. /*****************************************************************************
  38. * Operator-based class constructors
  39. *****************************************************************************/
  40. // read_ocr_class_cnn: Read an CNN-based OCR classifier from a file.
  41. explicit HOCRCnn(const HString& FileName);
  42. // read_ocr_class_cnn: Read an CNN-based OCR classifier from a file.
  43. explicit HOCRCnn(const char* FileName);
  44. #ifdef _WIN32
  45. // read_ocr_class_cnn: Read an CNN-based OCR classifier from a file.
  46. explicit HOCRCnn(const wchar_t* FileName);
  47. #endif
  48. /***************************************************************************
  49. * Operators *
  50. ***************************************************************************/
  51. // Clear an CNN-based OCR classifier.
  52. static void ClearOcrClassCnn(const HOCRCnnArray& OCRHandle);
  53. // Clear an CNN-based OCR classifier.
  54. void ClearOcrClassCnn() const;
  55. // Deserialize a serialized CNN-based OCR classifier.
  56. void DeserializeOcrClassCnn(const HSerializedItem& SerializedItemHandle);
  57. // Classify multiple characters with an CNN-based OCR classifier.
  58. HTuple DoOcrMultiClassCnn(const HRegion& Character, const HImage& Image, HTuple* Confidence) const;
  59. // Classify multiple characters with an CNN-based OCR classifier.
  60. HString DoOcrMultiClassCnn(const HRegion& Character, const HImage& Image, double* Confidence) const;
  61. // Classify a single character with an CNN-based OCR classifier.
  62. HTuple DoOcrSingleClassCnn(const HRegion& Character, const HImage& Image, const HTuple& Num, HTuple* Confidence) const;
  63. // Classify a single character with an CNN-based OCR classifier.
  64. HString DoOcrSingleClassCnn(const HRegion& Character, const HImage& Image, const HTuple& Num, double* Confidence) const;
  65. // Classify a related group of characters with an CNN-based OCR classifier.
  66. HTuple DoOcrWordCnn(const HRegion& Character, const HImage& Image, const HString& Expression, Hlong NumAlternatives, Hlong NumCorrections, HTuple* Confidence, HString* Word, double* Score) const;
  67. // Classify a related group of characters with an CNN-based OCR classifier.
  68. HString DoOcrWordCnn(const HRegion& Character, const HImage& Image, const HString& Expression, Hlong NumAlternatives, Hlong NumCorrections, double* Confidence, HString* Word, double* Score) const;
  69. // Classify a related group of characters with an CNN-based OCR classifier.
  70. HString DoOcrWordCnn(const HRegion& Character, const HImage& Image, const char* Expression, Hlong NumAlternatives, Hlong NumCorrections, double* Confidence, HString* Word, double* Score) const;
  71. #ifdef _WIN32
  72. // Classify a related group of characters with an CNN-based OCR classifier.
  73. HString DoOcrWordCnn(const HRegion& Character, const HImage& Image, const wchar_t* Expression, Hlong NumAlternatives, Hlong NumCorrections, double* Confidence, HString* Word, double* Score) const;
  74. #endif
  75. // Return the parameters of a CNN-based OCR classifier.
  76. HTuple GetParamsOcrClassCnn(const HTuple& GenParamName) const;
  77. // Return the parameters of a CNN-based OCR classifier.
  78. HTuple GetParamsOcrClassCnn(const HString& GenParamName) const;
  79. // Return the parameters of a CNN-based OCR classifier.
  80. HTuple GetParamsOcrClassCnn(const char* GenParamName) const;
  81. #ifdef _WIN32
  82. // Return the parameters of a CNN-based OCR classifier.
  83. HTuple GetParamsOcrClassCnn(const wchar_t* GenParamName) const;
  84. #endif
  85. // Get the names of the parameters that can be used in get_params_ocr_class_cnn for a given CNN-based OCR classifier.
  86. HTuple QueryParamsOcrClassCnn() const;
  87. // Read an CNN-based OCR classifier from a file.
  88. void ReadOcrClassCnn(const HString& FileName);
  89. // Read an CNN-based OCR classifier from a file.
  90. void ReadOcrClassCnn(const char* FileName);
  91. #ifdef _WIN32
  92. // Read an CNN-based OCR classifier from a file.
  93. void ReadOcrClassCnn(const wchar_t* FileName);
  94. #endif
  95. // Serialize a CNN-based OCR classifier
  96. HSerializedItem SerializeOcrClassCnn() const;
  97. };
  98. // forward declarations and types for internal array implementation
  99. template<class T> class HSmartPtr;
  100. template<class T> class HHandleBaseArrayRef;
  101. typedef HHandleBaseArrayRef<HOCRCnn> HOCRCnnArrayRef;
  102. typedef HSmartPtr< HOCRCnnArrayRef > HOCRCnnArrayPtr;
  103. // Represents multiple tool instances
  104. class LIntExport HOCRCnnArray : public HHandleBaseArray
  105. {
  106. public:
  107. // Create empty array
  108. HOCRCnnArray();
  109. // Create array from native array of tool instances
  110. HOCRCnnArray(HOCRCnn* classes, Hlong length);
  111. // Copy constructor
  112. HOCRCnnArray(const HOCRCnnArray &tool_array);
  113. // Destructor
  114. virtual ~HOCRCnnArray();
  115. // Assignment operator
  116. HOCRCnnArray &operator=(const HOCRCnnArray &tool_array);
  117. // Clears array and all tool instances
  118. virtual void Clear();
  119. // Get array of native tool instances
  120. const HOCRCnn* Tools() const;
  121. // Get number of tools
  122. virtual Hlong Length() const;
  123. // Create tool array from tuple of handles
  124. virtual void SetFromTuple(const HTuple& handles);
  125. // Get tuple of handles for tool array
  126. virtual HTuple ConvertToTuple() const;
  127. protected:
  128. // Smart pointer to internal data container
  129. HOCRCnnArrayPtr *mArrayPtr;
  130. };
  131. }
  132. #endif