From 68df84af8ac6a97175afd279a15ada0571948f43 Mon Sep 17 00:00:00 2001
From: Junru Li <lijunru@bytedance.com>
Date: Mon, 6 May 2024 06:25:25 +0000
Subject: [PATCH] solve issue #86
---
source/Lib/CommonLib/Buffer.cpp | 2 +-
source/Lib/CommonLib/NNFilterUnified.cpp | 4 +++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/source/Lib/CommonLib/Buffer.cpp b/source/Lib/CommonLib/Buffer.cpp
index 905d18bb22..b691f54e41 100644
--- a/source/Lib/CommonLib/Buffer.cpp
+++ b/source/Lib/CommonLib/Buffer.cpp
@@ -787,7 +787,7 @@ PelBuf PelStorage::getBuf( const CompArea &blk )
{
const PelBuf& r = bufs[blk.compID];
- //CHECKD( rsAddr( blk.bottomRight(), r.stride ) >= ( ( r.height - 1 ) * r.stride + r.width ), "Trying to access a buf outside of bound!" );
+ CHECKD( rsAddr( blk.bottomRight(), r.stride ) >= ( ( r.height - 1 ) * r.stride + r.width ), "Trying to access a buf outside of bound!" );
return PelBuf( r.buf + rsAddr( blk, r.stride ), r.stride, blk );
}
diff --git a/source/Lib/CommonLib/NNFilterUnified.cpp b/source/Lib/CommonLib/NNFilterUnified.cpp
index 0508622044..77af3f4f0b 100644
--- a/source/Lib/CommonLib/NNFilterUnified.cpp
+++ b/source/Lib/CommonLib/NNFilterUnified.cpp
@@ -589,7 +589,9 @@ void NNFilterUnified::filterBlock(Picture &pic, UnitArea inferArea, int extLeft,
NNInference::infer<TypeSadlLFUnified>(model, m_inputs);
- PelUnitBuf bufDst = m_scaled[0][prmId].getBuf(inferArea);
+ UnitArea inferAreaNoExt(inferArea.chromaFormat, Area(inferArea.lx() + extLeft, inferArea.ly() + extTop, inferArea.lwidth() - extLeft - extRight, inferArea.lheight() - extTop - extBottom));
+ UnitArea InferAreaExt(inferArea.chromaFormat, Area(-extLeft, -extTop, inferArea.lwidth(), inferArea.lheight()));
+ PelUnitBuf bufDst = m_scaled[0][prmId].getBuf(inferAreaNoExt).subBuf(InferAreaExt);
#if JVET_AH0080_TRANS_INPUT
if (m_nnlfTransInput) {
--
GitLab