update

xuepengqiang

2020-01-09 26d46c6b81c6936b89e1d3ab1e212417dbb23712

 reid_feature.cpp

@@ -34,7 +34,7 @@

}

bool ReID_Feature::ReID_extractor(unsigned char *pBuf, float *pFeature)
bool ReID_Feature::ReID_extractor(float *pBuf, float *pFeature)
{
    auto input_tensor = torch::from_blob(pBuf, {1, 256, 128, 3});
    input_tensor = input_tensor.permute({0, 3, 1, 2});