From b2500a8eb6665ce6efe0a7d954b6f101af83d7ec Mon Sep 17 00:00:00 2001 From: 554325746@qq.com <554325746@qq.com> Date: 星期三, 25 三月 2020 11:31:41 +0800 Subject: [PATCH] debug --- csrc/reid.cpp | 17 ++++++----------- 1 files changed, 6 insertions(+), 11 deletions(-) diff --git a/csrc/reid.cpp b/csrc/reid.cpp index 0adcb11..3b34bf4 100644 --- a/csrc/reid.cpp +++ b/csrc/reid.cpp @@ -4,20 +4,15 @@ #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) { if(init(gpu_index, module_path) < 0){ - throw std::runtime_error("init swscale error"); - } + throw std::runtime_error("init swscale error"); + } } reid::~reid() @@ -130,9 +125,9 @@ } } - for (int k = 0; k < 3; ++k) { - printf("--extractor---human_feats------%f",feature[k+2000]); - } + // for (int k = 0; k < 3; ++k) { + // printf("--extractor---human_feats------%f",feature[k+2000]); + // } printf("\n\n\n\n\n----------------------------------------------------------------------------\n"); -- Gitblit v1.8.0