Skip to content

Commit edecefe

Browse files
Update main.cpp (wang-xinyu#1167)
1 parent ad7e305 commit edecefe

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

yolov7/main.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
1-
#include "config.h"
1+
#include "config.h"
22
#include "model.h"
33
#include "cuda_utils.h"
44
#include "logging.h"
55
#include "utils.h"
66
#include "preprocess.h"
77
#include "postprocess.h"
88
#include <chrono>
9+
#include <fstream>
910

1011
using namespace nvinfer1;
1112

0 commit comments

Comments
 (0)