thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix cudaError_t error

Browse Source
This commit is contained in:
Marcos Luciano
2023-02-02 09:56:33 -03:00
parent 70c18a72a6
commit cd3ab90867
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
nvdsinfer_custom_impl_Yolo/calibrator.h
View File

@@ -7,6 +7,7 @@
#define CALIBRATOR_H #define CALIBRATOR_H
#include <vector> #include <vector>
#include <cuda_runtime_api.h>
#include "NvInfer.h" #include "NvInfer.h"
#include "opencv2/opencv.hpp" #include "opencv2/opencv.hpp"