#?將原始圖片轉換成需要的大小，并將其保存
#?========================================================================================
import?os
import?tensorflow?as?tf
from?PIL?import?Image

#?原始圖片的存儲位置
orig_picture?=?‘D:/ML/flower/flower_photos/‘

#?生成圖片的存儲位置
gen_picture?=?‘D:/ML/flower/input_data/‘

#?需要的識別類型
classes?=?{‘dandelion‘?‘roses‘?‘sunflowers‘‘tulips‘}

#?樣本總數
num_samples?=?4000


#?制作TFRecords數據
def?create_record（）:
????writer?=?tf.python_io.TFRecordWriter（“flower_train.tfrecords“）
????for?index?name?in?enumerate（classes）:
????????class_path?=?orig_picture?+?“/“?+?name?+?“/“
????????for?img_name?in?os.listdir（class_path）:
????????????img_path?=?class_path?+?img_name
????????????img?=?Image.open（img_path）
????????????img?=?img.resize（（64?64））??#?設置需要轉換的圖片大小
????????????img_raw?=?img.tobytes（）??#?將圖片轉化為原生bytes
????????????print（index?img_raw）
????????????example?=?tf.train.Example（
????????????????features=tf.train.Features（feature={
????????????????????“label“:?tf.train.Feature（int64_list=tf.train.Int64List（value=[index]））
????????????????????‘img_raw‘:?tf.train.Feature（bytes_list=tf.train.BytesList（value=[img_raw]））
????????????????}））
????????????writer.write（example.SerializeToString（））
????writer.close（）


#?=======================================================================================
def?read_and_decode（filename）:
????#?創建文件隊列不限讀取的數量
????filename_queue?=?tf.train.string_input_producer（[filename]）
????#?create?a?reader?from?file?queue
????reader?=?tf.TFRecordReader（）
????#?reader從文件隊列中讀入一個序列化的樣本
????_?serialized_example?=?reader.read（filename_queue）
????#?get?feature?from?serialized?example
????#?解析符號化的樣本
????features?=?tf.parse_single_example（
????????serialized_example
????????features={
????????????‘label‘:?tf.FixedLenFeature（[]?tf.int64）
????????????‘img_raw‘:?tf.FixedLenFeature（[]?tf.string）
????????}）
????label?=?features[‘label‘]
????img?=?features[‘img_raw‘]
????img?=?tf.decode_raw（img?tf.uint8）
????img?=?tf.reshape（img?[64?64?3]）
????#?img?=?tf.cast（img?tf.float32）?*?（1.?/?255）?-?0.5
????label?=?tf.cast（label?tf.int32）
????return?img?label


#?=======================================================================================
if?__name__?==?‘__main__‘:
????create_record（）
????batch?=?read_and_decode（‘flower_train.tfrecords‘）
????init_op?=?tf.group（tf.global_variables_initializer（）?tf.local_variables_initializer（））

????with?tf.Session（）?as?sess:??#?開始一個會話
????????sess.run（init_op）
????????coord?=?tf.train.Coordinator（）
????????threads?=?tf.train.start_queue_runners（coord=coord）

????????for?i?in?range（num_samples）:
????????????example?lab?=?sess.run（batch）??#?在會話中取出image和label
????????????img?=?Image.fromarray（example?‘RGB‘）??#?這里Image是之前提到的
????????????img.save（gen_picture?+?‘/‘?+?str（i）?+?‘samples‘?+?str（lab）?+?‘.jpg‘）??#?存下圖片;注意cwd后邊加上‘/’
????????????print（example?lab）
????????coord.request_stop（）
????????coord.join（threads）
????????sess.close（）