Fixed Stabilizer and created Object Detector tests

This commit is contained in:
Brenno
2020-07-30 21:39:20 -03:00
parent 58d2e8fe07
commit 6ca35bbf9b
13 changed files with 351 additions and 219 deletions

View File

@@ -110,13 +110,14 @@ class CVObjectDetection{
/// Protobuf Save and Load methods
// Save protobuf file
bool SaveTrackedData();
bool SaveObjDetectedData();
// Add frame object detection data into protobuf message.
void AddFrameDataToProto(libopenshotobjdetect::Frame* pbFrameData, CVDetectionData& dData);
// Load protobuf file
bool LoadTrackedData();
/// Get and Set JSON methods
void SetJson(const std::string value); ///< Load JSON string into this object
void SetJsonValue(const Json::Value root); ///< Load Json::Value into this object
// Load protobuf file (ONLY FOR MAKE TEST)
bool _LoadObjDetectdData();
};

View File

@@ -84,6 +84,7 @@ class CVStabilization {
size_t start;
size_t end;
double avr_dx, avr_dy, avr_da, max_dx, max_dy, max_da;
cv::Mat last_T;
cv::Mat prev_grey;
@@ -120,8 +121,6 @@ class CVStabilization {
bool SaveStabilizedData();
// Add frame stabilization data into protobuf message
void AddFrameDataToProto(libopenshotstabilize::Frame* pbFrameData, CamTrajectory& trajData, TransformParam& transData, size_t frame_number);
// Load protobuf data file
bool LoadStabilizedData();
// Return requested struct info for a given frame
TransformParam GetTransformParamData(size_t frameId);
@@ -131,6 +130,8 @@ class CVStabilization {
void SetJson(const std::string value); ///< Load JSON string into this object
void SetJsonValue(const Json::Value root); ///< Load Json::Value into this object
// Load protobuf data file (ONLY FOR MAKE TEST)
bool _LoadStabilizedData();
};
#endif

View File

@@ -98,12 +98,13 @@ class CVTracker {
bool SaveTrackedData();
// Add frame tracked data into protobuf message.
void AddFrameDataToProto(libopenshottracker::Frame* pbFrameData, FrameData& fData);
// Load protobuf file
bool LoadTrackedData();
/// Get and Set JSON methods
void SetJson(const std::string value); ///< Load JSON string into this object
void SetJsonValue(const Json::Value root); ///< Load Json::Value into this object
// Load protobuf file (ONLY FOR MAKE TEST)
bool _LoadTrackedData();
};