Update main.cpp (#1167)

2022-12-09 08:04:37 -05:00 · 2022-12-09 08:04:37 -05:00 · edecefe4c7
commit edecefe4c7
parent ad7e3052f4
1 changed files with 2 additions and 1 deletions
--- a/yolov7/main.cpp
+++ b/yolov7/main.cpp
@ -1,4 +1,4 @@
-#include "config.h"
+#include "config.h"
 #include "model.h"
 #include "cuda_utils.h"
 #include "logging.h"
@ -6,6 +6,7 @@
 #include "preprocess.h"
 #include "postprocess.h"
 #include <chrono>
+#include <fstream>

 using namespace nvinfer1;