From e51eb4d9657b30c1049282059a0aa74f7dce1e00 Mon Sep 17 00:00:00 2001
From: Thierry Dumas <thierry.dumas@InterDigital.com>
Date: Tue, 14 Feb 2023 09:49:06 +0100
Subject: [PATCH] Fix for issue 20
---
training/training_intra_prediction/generating_data_intra.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/training/training_intra_prediction/generating_data_intra.py b/training/training_intra_prediction/generating_data_intra.py
index 12929068da..a23cff0047 100644
--- a/training/training_intra_prediction/generating_data_intra.py
+++ b/training/training_intra_prediction/generating_data_intra.py
@@ -333,7 +333,7 @@ class GeneratorDataPart(object):
elif input_chroma_format == '444':
dict_input['data_add_basic'][:, :, 1].flatten().tofile(file)
dict_input['data_add_basic'][:, :, 2].flatten().tofile(file)
- args_encoding = [dict_picked['path_to_exe_encoder'],'-c',dict_picked['path_to_cfg'],'-i',dict_chr_own['pbfe'],'-b',dict_picked['path_to_bitstream'],'-wdt',str(width_video),'-hgt',str(height_video),'--InputBitDepth={}'.format(input_bit_depth),'--OutputBitDepth={}'.format(input_bit_depth),'--InputChromaFormat={}'.format(input_chroma_format),'--SEIDecodedPictureHash=1','--ReconFile=','--FramesToBeEncoded={}'.format(nb_frames),'--FrameSkip={}'.format(nb_fixed),'--QP={}'.format(numpy.random.choice(dict_picked['qps_int']).item())]
+ args_encoding = [dict_picked['path_to_exe_encoder'],'-c',dict_picked['path_to_cfg'],'-i',dict_chr_own['pbfe'],'-b',dict_picked['path_to_bitstream'],'-wdt',str(width_video),'-hgt',str(height_video),'--InputBitDepth={}'.format(input_bit_depth),'--OutputBitDepth={}'.format(input_bit_depth),'--InputChromaFormat={}'.format(input_chroma_format),'--SEIDecodedPictureHash=1','--UpscaledOutput=0','--ReconFile=','--FramesToBeEncoded={}'.format(nb_frames),'--FrameSkip={}'.format(nb_fixed),'--QP={}'.format(numpy.random.choice(dict_picked['qps_int']).item())]
if dict_picked['descr_stacked_up']:
args_encoding.append('--DescriptionPairHeightWidthPathToGraphOutput={}'.format(dict_picked['descr_stacked_up']))
args_encoding.append('--PrefixAbsolutePathsToGraphsOutput=')
@@ -351,7 +351,7 @@ class GeneratorDataPart(object):
if are_orgf_definitely[1]:
args_encoding.append('--AreOriginalFh=1')
del are_orgf_definitely
- args_decoding = [dict_picked['path_to_exe_decoder'],'-b',dict_picked['path_to_bitstream'],'-o',dict_picked['path_provided_2'],'--OutputBitDepth={}'.format(input_bit_depth)]
+ args_decoding = [dict_picked['path_to_exe_decoder'],'-b',dict_picked['path_to_bitstream'],'-o',dict_picked['path_provided_2'],'--OutputBitDepth={}'.format(input_bit_depth),'--UpscaledOutput=0']
if dict_picked['descr_stacked_up']:
args_decoding.append('--DescriptionPairHeightWidthPathToGraphOutput={}'.format(dict_picked['descr_stacked_up']))
args_decoding.append('--PrefixAbsolutePathsToGraphsOutput=')
--
GitLab