Hi, let me make sure I have it right. We need generate the "two" tflie file(output size are [1, 256, 265, 3] and [1, 1088, 1920, 3]) base on the same ckpt file, and than use '[1, 256, 265, 3]' tflite to generate the 256x256 RGB image. Finally summit the RGB image and [1, 1088, 1920, 3] tflite...