uptate utils

xuepengqiang

2020-01-09 b7e0f4cc5d5eef69ba668eefa67b46504570b7ba

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