update

zhangmeng

2020-01-14 5459ba1d3f7f944aa97923ed9c09a5dbc7663928

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