diff --git a/models/object_detection_yolox/demo.cpp b/models/object_detection_yolox/demo.cpp index 35319b60..074e02f4 100644 --- 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);