Add YOLOX C++ demo save functionality (#235)

* Add saving functionality to C++ demo * Updated saving to image inputs only * Removed unnecessary bool
opencv · Feb 28, 2024 · 488fb8e · 488fb8e
1 parent f53754a
commit 488fb8e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/models/object_detection_yolox/demo.cpp b/models/object_detection_yolox/demo.cpp
@@ -306,6 +306,10 @@ int main(int argc, char** argv)
         tm.stop();
         cout << "Inference time: " << tm.getTimeMilli() << " ms\n";
         Mat img = visualize(predictions, frame, letterboxScale, tm.getFPS());
+        if (save && parser.has("input"))
+        {
+            imwrite("result.jpg", img);
+        }
         if (vis)
         {
             imshow(kWinName, img);