Skip to content

Commit 6defb2b

Browse files
committed
hrnet classification trt 加速
1 parent 4d01e09 commit 6defb2b

File tree

2 files changed

+1
-1164
lines changed

2 files changed

+1
-1164
lines changed

HRNetClassification/hrnet.cpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,11 +10,8 @@
1010
#include "NvInferPlugin.h"
1111
#include "cuda_runtime_api.h"
1212
#include "logging.h"
13-
#ifdef WIN32
14-
#include "include/dirent.h" // WIN32
15-
#else
1613
#include <dirent.h> // LINUX
17-
#endif
14+
1815

1916

2017
using namespace nvinfer1;

0 commit comments

Comments
 (0)