From 9c0015909e0c655309440a496d369f13dfce9a1e Mon Sep 17 00:00:00 2001
From: Liqiang Wang <liqiangwang@tencent.com>
Date: Fri, 12 Aug 2022 14:56:55 +0800
Subject: [PATCH] A small bugfix of common inference API to harmonize with
JVET-AA0088
---
source/Lib/CommonLib/NNInference.h | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/source/Lib/CommonLib/NNInference.h b/source/Lib/CommonLib/NNInference.h
index ad72de96a4..257584619e 100644
--- a/source/Lib/CommonLib/NNInference.h
+++ b/source/Lib/CommonLib/NNInference.h
@@ -132,7 +132,7 @@ public:
#endif
}
template<typename T>
- static void prepareInputs (Picture* pic, UnitArea inferArea, vector<sadl::Tensor<T>> &inputs, int globalQp, int localQp, bool inter, std::vector<InputData> listInputData)
+ static void prepareInputs (Picture* pic, UnitArea inferArea, vector<sadl::Tensor<T>> &inputs, int globalQp, int localQp, int sliceType, std::vector<InputData> listInputData)
{
for (auto &inputData : listInputData)
{
@@ -157,7 +157,7 @@ public:
fillInputFromConstant<T>(pic, inferArea, inputs[inputData.index], localQp, inputData.luma, inputData.scale, inputData.shift);
break;
case NN_INPUT_SLICE_TYPE:
- fillInputFromConstant<T>(pic, inferArea, inputs[inputData.index], inter ? 1 : 0, inputData.luma, inputData.scale, inputData.shift);
+ fillInputFromConstant<T>(pic, inferArea, inputs[inputData.index], sliceType, inputData.luma, inputData.scale, inputData.shift);
break;
default:
THROW("invalid input data");
@@ -170,7 +170,7 @@ public:
{
if (!model.apply(inputs))
{
- cerr << "[ERROR] issue during luma model inference" << endl;
+ cerr << "[ERROR] issue during inference" << endl;
exit(-1);
}
}
--
GitLab