FLImaging 6.10.14.2
ValidatorForObjectDetection.h
1#pragma once
2
3#if _MSC_VER >= 1900 && defined(_M_X64)
4
5#include "ValidatorBase.h"
6
7namespace FLImaging
8{
9 namespace AI
10 {
11 template<typename T>
12 class CObjectBox;
13
14 template <typename T>
15 class CDetectionInfo
16 {
17 public:
18 int64_t i64UniqueTruthIndex;
19 int64_t i64TruthFlag;
20 int64_t i64ClassId;
21 T tIoU;
22 T tScore;
23
24 public:
25 CDetectionInfo();
26 };
27
28 template <typename T>
29 class CPrecisionRecall
30 {
31 public:
32 int64_t i64TP;
33 int64_t i64FP;
34 int64_t i64FN;
35 T tScore;
36 double f64Precision;
37 double f64Recall;
38
39 public:
40 CPrecisionRecall();
41 };
42
43 template <typename T>
44 class FL_EXPORT CValidatorForObjectDetection : public CValidatorBase<T>
45 {
46 public:
47 enum EModel
48 {
49 EModel_RFLNet = 0,
50 };
51
52 public:
53 CValidatorForObjectDetection();
54 CValidatorForObjectDetection(const CValidatorForObjectDetection<T>& validator);
55 CValidatorForObjectDetection(const CValidatorForObjectDetection<T>* pValidator);
56 virtual ~CValidatorForObjectDetection();
57
58 const CValidatorForObjectDetection<T>& operator=(const CValidatorForObjectDetection<T>& validator);
59
60 virtual const CResult Assign(const CValidatorForObjectDetection<T>& validator);
61 virtual const CResult Assign(const CValidatorForObjectDetection<T>* pValidator);
62
63 virtual const CResult Clear() override;
64 virtual const CResult ClearResultBox();
65
66 virtual const CResult SetNumOfClasses(int64_t i64ClassCount);
67 virtual int64_t GetNumOfClasses() const;
68
69 virtual const CResult SetNonMaximumSuppressionType(ENonMaximumSuppressionType eNMS = ENonMaximumSuppressionType_Greedy);
70 virtual ENonMaximumSuppressionType GetNonMaximumSuppressionType() const;
71 virtual const CResult SetBetaNMS(T tBetaNMS = .6);
72 virtual T GetBetaNMS() const;
73 virtual const CResult SetNMSIoUThreshold(T tNMSIoUThreshold = 0.45);
74 virtual T GetNMSIoUThreshold() const;
75 virtual const CResult SetObjectnessThreshold(T tObjectnessThreshold = .005);
76 virtual T GetObjectnessThreshold() const;
77
78 virtual const CResult SetIoUThreshold(T tIoUThreshold = .5);
79 virtual T GetIoUThreshold() const;
80 virtual const CResult SetAverageIoUThreshold(T tCalcAvgIoUThreshold = .25);
81 virtual T GetAverageIoUThreshold() const;
82
83 virtual const CResult SetRealClassIdToValidClassId(const std::vector<int64_t>& vctRealClassId);
84 virtual const CResult GetRealClassIdToValidClassId(std::vector<int64_t>& vctRealClassId);
85
86 virtual const std::vector<std::vector<std::vector<std::tuple<CObjectBox<T>, T, int64_t>>>>* GetResultBox() const;
87
88 virtual T Validate(int32_t* pI32IterationFeedback = nullptr) override;
89
90 static T CalculateMap(std::vector<CDetectionInfo<T>>& vctDetectionInfo, const int64_t* pI64TruthClassesCount, int64_t i64UniqueTruthCount, int64_t i64ClassCount, std::vector<T>& vctResultPrecisionPerClass, std::vector<T>& vctResultRecallPerClass, std::vector<T>& vctResultAPClass);
91
92 static const CResult CalculateEvaluation(const std::vector<CDetectionInfo<T>>& vctDetectionInfo, const int64_t* pI64TruthClassesCount, int64_t i64UniqueTruthCount, int64_t i64ClassCount, Base::CFLArray<uint8_t>& flaResultCorrectness, Base::CFLArray<double>& flaResultIoU, Base::CFLArray<int64_t>& flaFalseNegativeTargetIndexes);
93
94 static const CResult MakeDetectionInfo(const Foundation::CFLFigureArray& flfa2Figure, const Base::CFLArray<Base::CFLArray<int32_t>>& fla2ClassNum, const Base::CFLArray<Base::CFLArray<double>> fla2Score, const Foundation::CFLFigureArray& flfa2Target, int64_t i64TotalClass, T tIoUThrehsold, std::vector<CDetectionInfo<T>>& vctResultDetectionInfoFlatten, std::vector<int64_t>& vctResultTruthClassesCountFlatten, int64_t& i64ResultUniqueTruthTotal);
95
96 virtual const CResult SetModel(EModel eModel);
97 virtual EModel GetModel() const;
98
99 virtual const CResult EnableIOUThresholdUpdate(bool bIOUThresholdUpdate = true);
100 virtual bool IsIOUThresholdUpdateEnabled() const;
101
102 virtual const CResult EnableAngleReflection(bool bAngleReflection = false);
103 virtual bool IsAngleReflectionEnabled() const;
104
105 virtual const CResult EnableSingleLabel(bool bSingleLabel);
106 virtual bool IsSingleLabelEnable() const;
107
108 virtual const std::vector<T>& GetRecallPerClass();
109 virtual const std::vector<T>& GetPrecisionPerClass();
110 virtual const std::vector<T>& GetAPPerClass();
111
112 DeclareGetClassType();
113 SupportToDuplicateObject(CValidatorForObjectDetection<T>, *this);
114
115 protected:
116 virtual const CResult ResetComputationPerClass();
117 virtual const CResult SetModelParams();
118 virtual const CResult SetModelParams(const Base::CFLData& fldData);
119 virtual const CResult BackupModelParams(Base::CFLData& fldData);
120
121 std::vector<int64_t>& m_vctRealClassIdToValidClassId;
122
123 ENonMaximumSuppressionType m_eValidationNMS;
124 T m_tValidationBetaNMS;
125 T m_tValidationNMSIoUThreshold;
126 T m_tValidationObjectnessThreshold;
127
128 T m_tValidationIoUThreshold;
129 T m_tValidationCalcAvgIoUThreshold;
130 T m_tLastValidation;
131
132 int64_t m_i64TruthLabelSize;
133 T* m_pTTruthLabels;
134
135 int64_t m_i64ClassCount;
136 int64_t* m_pI64TruthClassesCount;
137 int64_t* m_pI64TPForThresholdPerClass;
138 int64_t* m_pI64FPForThresholdPerClass;
139 T* m_pTAvgIoUPerClass;
140
141 EModel m_eModel;
142
143 bool m_bIOUThresholdUpdate;
144 bool m_bAngleReflection;
145 bool m_bSingleLabel;
146 std::vector<std::vector<std::vector<std::tuple<CObjectBox<T>, T, int64_t>>>>& m_vct3BoxResult;
147
148 std::vector<T>& m_vctPrecisionPerClass;
149 std::vector<T>& m_vctRecallPerClass;
150 std::vector<T>& m_vctAPPerClass;
151 };
152
153
154 typedef CValidatorForObjectDetection<float> CValidatorForObjectDetectionF;
155 typedef CValidatorForObjectDetection<double> CValidatorForObjectDetectionD;
156 }
157}
158
159#endif
FLImaging::AI
Definition AlgorithmAIBase.h:18
FLImagingCLR::GUI::EAvailableView3DFigureContextMenu::Assign
@ Assign
Set the value of CGUIPropertyItemView3DFigure to the specified figure.
Definition DefinitionsGUIView3D.h:2798
FLImagingCLR::GUI::EMenuItemObjectFigure::Clear
@ Clear
Clear all the figure objects.
Definition DefinitionsGUI.h:2092
Generated on for FLImaging by doxygen 1.14.0