openshot/libopenshot
0
0
Fork 0
mirror of https://github.com/OpenShot/libopenshot.git synced 2026-03-02 08:53:52 -08:00
Code Issues Packages Projects Releases Wiki Activity

Changed bounding box draw in Object Detector effect

Browse Source
This commit is contained in:
Brenno
2020-07-29 18:50:11 -03:00
parent 483f2883c9
commit a8d877cb3b
9 changed files with 80 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/CVObjectDetection.cpp
View File

@@ -166,17 +166,21 @@ void CVObjectDetection::postprocess(const cv::Size &frameDims, const std::vector
std::vector<int> indices;
cv::dnn::NMSBoxes(boxes, confidences, confThreshold, nmsThreshold, indices);
// std::vector<cv::Rect> sortBoxes;
// for(auto box : boxes)
// sortBoxes.push_back(box);
// sort.update(sortBoxes, frameId, sqrt(pow(frameDims.width,2) + pow(frameDims.height, 2)));
std::vector<cv::Rect> sortBoxes;
for(auto box : boxes)
sortBoxes.push_back(box);
sort.update(sortBoxes, frameId, sqrt(pow(frameDims.width,2) + pow(frameDims.height, 2)), confidences, classIds);
// sortBoxes.clear();
// for(auto TBox : sort.frameTrackingResult)
// if(TBox.frame == frameId){
// sortBoxes.push_back(TBox.box);
// }
sortBoxes.clear(); boxes.clear(); confidences.clear(); classIds.clear();
for(auto TBox : sort.frameTrackingResult){
if(TBox.frame == frameId){
boxes.push_back(TBox.box);
confidences.push_back(TBox.confidence);
classIds.push_back(TBox.classId);
}
}
// for(int i = 0; i<boxes.size(); i++){
// bool found = false;