Wie lade ich verschiedene Bilder mit `tf.train.string_input_prodecer()` im Tensorflow?

Question

Ich bekomme ein Bildset voller Bilder mit verschiedenen Größen und Erweiterungen. Die meisten ihrer Farbräume sind RGB, während einige den zusätzlichen Alpha-Kanal haben.

Wie kann ich sie konvertieren in 227 * 227 * 3 (3 bedeutet RGB) in Tensorflow?
Vielen Dank!

Answer 1

Werfen Sie einen Blick auf die Image APIs

Insbesondere was für Sie von Interesse ist, wäre vielleicht die folgenden Funktionen im Cropping-Header:

tf.image.resize_image_with_crop_or_pad 
tf.image.central_crop 
tf.image.pad_to_bounding_box 
tf.image.crop_to_bounding_box 
tf.image.extract_glimpse 
tf.image.crop_and_resize 

Wie für den Alpha-Kanal, es hängt wirklich wie du in den Bildern liest. Wenn Sie tf.image.decode_image verwenden, können Sie vielleicht versuchen, channels=3/4 zu spezifizieren (ich bin nicht sicher, ob das den Code bricht), aber wenn es wie das ist, was Sie erwähnten, haben einige Bilder Alpha und einige nicht, dann könnten Sie cv2 verwenden oder PIL, um die Bilder vollständig vorzubearbeiten und sie auf 227 x 227 x 3 zu beschneiden.