object detection

opencv-yolo/main.cpp

@@ -11,12 +11,14 @@ int main(int argc, char *argv[]){
     bool bVerbose = false;
     bool help = false;
     bool bCont = false;
+    int wrap = 10;
     string sInput, sOutput = "detect.jpg";
     string modelPath = ".";
 
     auto cli = (
         value("input path", sInput),
         option("-cl") & value("confidence level of detection, default: 0.1", fConfident),
+        option("-w", "--wrap") & value("output file wrap. defualt: 10; 0 - no wrap", wrap),
         option("-vv", "--debug").set(bVerbose).doc("verbose prints"),
         option("-human", "--human-only") & value("detect only human object, default: true", bHumanOnly),
         option("-c", "--config-path") & value("model and configuration path", modelPath),
@@ -31,12 +33,12 @@ int main(int argc, char *argv[]){
         spdlog::info(s.str());
         exit(0);
     }
-    spdlog::info("{} {}", bHumanOnly, fConfident);
+    spdlog::info("{} {} {}", bHumanOnly, fConfident, wrap);
 
     if(bVerbose) {
         spdlog::set_level(spdlog::level::debug);
     }
-
-    YoloDectect detector(modelPath, bHumanOnly == "true"?true:false, fConfident, bCont);
+ 
+    YoloDectect detector(modelPath, bHumanOnly == "true"?true:false, fConfident, bCont, wrap);
     detector.process(sInput, sOutput);
 }
\ No newline at end of file

opencv-yolo/yolo.hpp

@@ -226,7 +226,7 @@ public:
         }
 
         try{
-            if(inVideoUri.substr(inVideoUri.find_last_of(".") + 1) == "mp4"||(cameNo = stoi(inVideoUri)) >= 0) {
+            if(inVideoUri.substr(0, 4) == "rtsp"||inVideoUri.substr(0, 4) == "rtmp"||inVideoUri.substr(inVideoUri.find_last_of(".") + 1) == "mp4"||(cameNo = stoi(inVideoUri)) >= 0) {
                 bInputIsImage = false;
             }
         }catch(...) {
@@ -330,7 +330,7 @@ public:
                         if(wrapNum > 0) {
                             detCnt = detCnt % wrapNum;
                         }
-                        
+
                         string ofname = outFileBase + to_string(detCnt) + ".jpg";
                         imwrite(ofname, outFrame);
                         string msg = fmt::format("{} found {} {}:\n", selfId, ret.size(), ofname);