اشکال در تغییر اندازه تصویر در تنسورفلو

Question

سلام به همگی.

من با استفاده از کد زیر تصویر را resize می کنم ولی خروجی تصویر نویزی شده علت خرابی کجاست؟

   try:
      file = tf.read_file(file_name)
      image = tf.image.decode_jpeg(file)
   except:
      continue

   gray_img = tf.image.rgb_to_grayscale(image)
   norm_img = tf.image.resize_images(gray_img,size=[400,400])


   img_bytes =  sess.run( tf.image.convert_image_dtype(norm_img,dtype= tf.uint8))
   cv2.imshow("view",img_bytes)
   cv2.waitKey(0)

 

Answer 1

تابع resize_images تنسور مورد نظر را float می کنه به همین خاطر نیاز دارید که اون را مجدد به بایت تبدیل کنید برای اینکار از tf.cast استفاده کنید.

img_bytes =  sess.run( tf.cast(norm_img,dtype= tf.uint8))