csrc/reid.cpp
@@ -4,13 +4,8 @@ #include <stdexcept> // #include <cuda_runtime_api.h> #include <torch/torch.h> #include <opencv2/core/core.hpp> #include <opencv2/highgui/highgui.hpp> #include "opencv2/imgproc/imgproc.hpp" #include "opencv2/opencv.hpp" #include "opencv2/videoio/videoio.hpp" #include "opencv2/video.hpp" using namespace std; reid::reid(const int gpu_index, const char *module_path) :is_gpu_(gpu_index >= 0) ,start_(false)
