Please modify output_image_width and output_image_width to meet the requirement mentioned in Integrating TAO Models into DeepStream — TAO Toolkit 3.22.05 documentation
Input size : C * W * H (where C = 1 or 3, W >= 128, H >= 128, W, H are multiples of 32)