-
大小: 53.54MB文件類型: .zip金幣: 1下載: 0 次發布日期: 2022-05-24
- 語言: Python
- 標簽: tensorflow??mnist??python??
資源簡介
博客:https://blog.csdn.net/weixin_45775701/article/details/109446515
代碼片段和文件信息
import?os
import?glob
import?tensorflow?as?tf
import?mnist_inference
import?mnist_train
from?PIL?import?Image?ImageFilter
import?matplotlib.pyplot?as?plt
import?cv2
def?image_prepare(pic_name):
????#?讀取圖像,第二個參數是讀取方式
????img?=?cv2.imread(pic_name?1)
????#?使用全局閾值,降噪
????retth1?=?cv2.threshold(img127255cv2.THRESH_BINARY)
????#?把opencv圖像轉化為PIL圖像
????im?=?Image.fromarray(cv2.cvtColor(th1cv2.COLOR_BGR2RGB))
????#?灰度化
????im?=?im.convert(‘L‘)
????#?為圖片重新指定尺寸
????im?=?im.resize((2828)?Image.ANTIALIAS)
????#?plt.imshow(im)
????#?plt.show()
????#?圖像轉換為list
????im_list?=?list(im.getdata())
????#?圖像灰度反轉
????result?=?[(255-x)*1.0/255.0?for?x?in?im_list]
????return?result
def?main(argv=None):
????#?把要識別的圖片放到下面的文件夾中
????img_path?=?‘picture/‘
????imgs?=?glob.glob(os.path.join(img_path?‘*‘))
????for?p?in?imgs:
????????#?圖像處理:降噪、灰度化、修改尺寸以及灰度反轉
????????pic?=?image_prepare(p)
????????#?識別圖像
????????evaluate(pic?p)
def?evaluate(pic?pic_name):
????with?tf.Graph().as_default()?as?g:
????????x?=?tf.placeholder(tf.float32?[None?mnist_inference.INPUT_NODE]?name=‘x-input‘)
????????y_?=?tf.placeholder(tf.float32?[None?mnist_inference.OUTPUT_NODE]?name=‘y-input‘)
????????validate_feed?=?{x:[pic]}
????????y?=?mnist_inference.inference(x?None)
????????result?=?tf.argmax(y?1)
????????variable_averages?=?tf.train.ExponentialMovingAverage(mnist_train.MOVING_AVERAGE_DECAY)
????????variables_to_restore?=?variable_averages.variables_to_restore()
????????saver?=?tf.train.Saver(variables_to_restore)
????????with?tf.Session()?as?sess:
????????????ckpt?=?tf.train.get_checkpoint_state(mnist_train.MODEL_SAVE_PATH)
????????????if?ckpt?and?ckpt.model_checkpoint_path:
????????????????saver.restore(sess?ckpt.model_checkpoint_path)
????????????????global_step?=?ckpt.model_checkpoint_path.split(‘/‘)[-1].split(‘-‘)[-1]
????????????????number?=?sess.run(result?feed_dict=validate_feed)
????????????????pic_name?=?pic_name.split(‘\\‘)[-1]
????????????????print(pic_name‘?is?:‘number[0])
????????????else:
????????????????print(‘No?checkpoint?file?found‘)
????????????????return
if?__name__?==?‘__main__‘:
????main()
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2020-11-03?13:06??Handwritten-Numeral-Recognition\
?????目錄???????????0??2020-11-02?12:07??Handwritten-Numeral-Recognition\MNIST_data\
?????文件???????60008??2020-11-01?10:36??Handwritten-Numeral-Recognition\MNIST_data\train-labels.idx1-ubyte
?????目錄???????????0??2020-11-03?18:18??__MACOSX\
?????目錄???????????0??2020-11-03?18:18??__MACOSX\Handwritten-Numeral-Recognition\
?????目錄???????????0??2020-11-03?18:18??__MACOSX\Handwritten-Numeral-Recognition\MNIST_data\
?????文件?????????232??2020-11-01?10:36??__MACOSX\Handwritten-Numeral-Recognition\MNIST_data\._train-labels.idx1-ubyte
?????文件????47040016??2020-11-01?11:30??Handwritten-Numeral-Recognition\MNIST_data\train-images.idx3-ubyte
?????文件?????????232??2020-11-01?11:30??__MACOSX\Handwritten-Numeral-Recognition\MNIST_data\._train-images.idx3-ubyte
?????文件???????10008??2020-11-01?10:35??Handwritten-Numeral-Recognition\MNIST_data\t10k-labels.idx1-ubyte
?????文件?????????232??2020-11-01?10:35??__MACOSX\Handwritten-Numeral-Recognition\MNIST_data\._t10k-labels.idx1-ubyte
?????文件?????1648877??2020-11-01?11:41??Handwritten-Numeral-Recognition\MNIST_data\t10k-images-idx3-ubyte.gz
?????文件?????7840016??2020-11-01?10:36??Handwritten-Numeral-Recognition\MNIST_data\t10k-images.idx3-ubyte
?????文件?????????232??2020-11-01?10:36??__MACOSX\Handwritten-Numeral-Recognition\MNIST_data\._t10k-images.idx3-ubyte
?????文件?????9912422??2020-11-01?11:37??Handwritten-Numeral-Recognition\MNIST_data\train-images-idx3-ubyte.gz
?????文件???????28881??2020-11-01?11:37??Handwritten-Numeral-Recognition\MNIST_data\train-labels-idx1-ubyte.gz
?????文件????????4542??2020-11-01?11:45??Handwritten-Numeral-Recognition\MNIST_data\t10k-labels-idx1-ubyte.gz
?????文件?????????176??2020-11-01?11:45??__MACOSX\Handwritten-Numeral-Recognition\MNIST_data\._t10k-labels-idx1-ubyte.gz
?????文件????????1232??2019-06-05?15:21??Handwritten-Numeral-Recognition\mnist_inference.py
?????文件?????????274??2019-06-05?15:21??__MACOSX\Handwritten-Numeral-Recognition\._mnist_inference.py
?????文件????????2683??2020-11-01?11:44??Handwritten-Numeral-Recognition\mnist_train.py
?????文件?????????274??2020-11-01?11:44??__MACOSX\Handwritten-Numeral-Recognition\._mnist_train.py
?????文件???????32248??2020-11-01?20:08??Handwritten-Numeral-Recognition\timg.jpeg
?????文件?????????176??2020-11-01?20:08??__MACOSX\Handwritten-Numeral-Recognition\._timg.jpeg
?????文件???????10265??2020-11-03?12:37??Handwritten-Numeral-Recognition\as.png
?????文件?????????233??2020-11-03?12:37??__MACOSX\Handwritten-Numeral-Recognition\._as.png
?????文件???????14340??2020-11-03?13:06??Handwritten-Numeral-Recognition\.DS_Store
?????文件?????????120??2020-11-03?13:06??__MACOSX\Handwritten-Numeral-Recognition\._.DS_Store
?????目錄???????????0??2020-11-01?19:30??Handwritten-Numeral-Recognition\MNIST_model\
?????文件?????3180084??2020-11-01?19:29??Handwritten-Numeral-Recognition\MNIST_model\mnist_model-6001.data-00000-of-00001
?????文件?????3180084??2019-06-05?14:53??Handwritten-Numeral-Recognition\MNIST_model\mnist_model-28001.data-00000-of-00001
............此處省略116個文件信息
評論
共有 條評論