I suggest the "tf.image.convert_image_dtype” operation should be outside the loop, and use the placeholer, which leads to outmemory!

[donghaoye]

[xthan]

Hi Haoye, Thanks for your suggestion. I will have a look at it once I have some time.