From 26d46c6b81c6936b89e1d3ab1e212417dbb23712 Mon Sep 17 00:00:00 2001
From: xuepengqiang <506321815@qq.com>
Date: 星期四, 09 一月 2020 09:36:08 +0800
Subject: [PATCH] update
---
reid_feature.cpp | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/reid_feature.cpp b/reid_feature.cpp
index 4bb847e..fdf1e88 100644
--- a/reid_feature.cpp
+++ b/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});
--
Gitblit v1.8.0