#?將原始圖片轉(zhuǎn)換成需要的大小，并將其保存
#?========================================================================================
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‘}

#?樣本總數(shù)
num_samples?=?4000


#?制作TFRecords數(shù)據(jù)
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））??#?設置需要轉(zhuǎn)換的圖片大小
????????????img_raw?=?img.tobytes（）??#?將圖片轉(zhuǎn)化為原生bytes
????????????print（index?img_raw）
????????????example?=?tf.train.Example（
????????????????features=tf.train.Features（feature={
????????????????????“l(fā)abel“:?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）:
????#?創(chuàng)建文件隊列不限讀取的數(shù)量
????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（）

?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-08-23?01:45??flower_world-master\
?????目錄???????????0??2018-08-23?01:45??flower_world-master\.idea\
?????文件?????????262??2018-08-23?01:45??flower_world-master\.idea\misc.xml
?????文件?????????270??2018-08-23?01:45??flower_world-master\.idea\modules.xml
?????文件?????????186??2018-08-23?01:45??flower_world-master\.idea\other.xml
?????文件?????????398??2018-08-23?01:45??flower_world-master\.idea\untitled9.iml
?????文件???????41471??2018-08-23?01:45??flower_world-master\.idea\workspace.xml
?????文件????????1063??2018-08-23?01:45??flower_world-master\LICENSE
?????文件????????1099??2018-08-23?01:45??flower_world-master\README.md
?????目錄???????????0??2018-08-23?01:45??flower_world-master\__pycache__\
?????文件????????2601??2018-08-23?01:45??flower_world-master\__pycache__\input_data.cpython-35.pyc
?????文件????????3420??2018-08-23?01:45??flower_world-master\__pycache__\model.cpython-35.pyc
?????文件????????2316??2018-08-23?01:45??flower_world-master\__pycache__\test.cpython-35.pyc
?????文件????????3223??2018-08-23?01:45??flower_world-master\create?record.py
?????文件????????2970??2018-08-23?01:45??flower_world-master\gui.py
?????文件????????4428??2018-08-23?01:45??flower_world-master\input_data.py
?????文件?????6263437??2018-08-23?01:45??flower_world-master\input_data.rar
?????文件????????6429??2018-08-23?01:45??flower_world-master\model.py
?????文件????????2531??2018-08-23?01:45??flower_world-master\test.py
?????文件????????2768??2018-08-23?01:45??flower_world-master\train.py