debug

554325746@qq.com

2020-03-25 b2500a8eb6665ce6efe0a7d954b6f101af83d7ec

 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");