Skip to content

Commit 1c4c09c

Browse files
authored
AC: handle output renaming for background matting (#3288)
1 parent 533563e commit 1c4c09c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/adapters/background_matting.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@ def process(self, raw, identifiers, frame_meta):
3636
batch_size = len(identifiers)
3737
for i in range(batch_size):
3838
output = {
39-
self.pha: self.to_image(pha[i], frame_meta[i]),
40-
self.fgr: self.to_image(fgr[i], frame_meta[i])
39+
self._orig_pha: self.to_image(pha[i], frame_meta[i]),
40+
self._orig_fgr: self.to_image(fgr[i], frame_meta[i])
4141
}
4242
result.append(
4343
BackgroundMattingPrediction(identifiers[i], output)
@@ -77,5 +77,7 @@ def select_output_blob(self, outputs):
7777

7878
def configure(self):
7979
self.pha = self.get_value_from_config('alpha_out')
80+
self._orig_pha = self.get_value_from_config('alpha_out')
8081
self.fgr = self.get_value_from_config('foreground_out')
82+
self._orig_fgr = self.get_value_from_config('foreground_out')
8183
self.output_verified = False

0 commit comments

Comments
 (0)