資源簡介
最經典的VGG代碼，是深度學習的基礎，適合配合VGG代碼一起閱讀使用，非常適合深度學習初學者閱讀學習深度學習框架構建規則。
代碼片段和文件信息
import?numpy?as?np
import?tensorflow?as?tf
import?vgg16
import?utils
img1?=?utils.load_image（“./test_data/tiger.jpeg“）
img2?=?utils.load_image（“./test_data/puzzle.jpeg“）
batch1?=?img1.reshape（（1?224?224?3））
batch2?=?img2.reshape（（1?224?224?3））
batch?=?np.concatenate（（batch1?batch2）?0）
#?with?tf.Session（config=tf.ConfigProto（gpu_options=（tf.GPUOptions（per_process_gpu_memory_fraction=0.7））））?as?sess:
with?tf.device（‘/cpu:0‘）:
????with?tf.Session（）?as?sess:
????????images?=?tf.placeholder（“float“?[2?224?224?3]）
????????feed_dict?=?{images:?batch}
????????vgg?=?vgg16.Vgg16（）
????????with?tf.name_scope（“content_vgg“）:
????????????vgg.build（images）
????????prob?=?sess.run（vgg.prob?feed_dict=feed_dict）
????????print（prob）
????????utils.print_prob（prob[0]?‘./synset.txt‘）
??????屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2017-06-12?08:54??tensorflow-vgg-master\
?????文件??????????76??2017-06-12?08:54??tensorflow-vgg-master\.gitignore
?????文件????????2330??2017-06-12?08:54??tensorflow-vgg-master\README.md
?????文件???????????0??2017-06-12?08:54??tensorflow-vgg-master\__init__.py
?????文件???????31675??2017-06-12?08:54??tensorflow-vgg-master\synset.txt
?????目錄???????????0??2017-06-12?08:54??tensorflow-vgg-master\test_data\
?????文件????????8892??2017-06-12?08:54??tensorflow-vgg-master\test_data\puzzle.jpeg
?????文件???????12270??2017-06-12?08:54??tensorflow-vgg-master\test_data\tiger.jpeg
?????文件?????????845??2017-06-12?08:54??tensorflow-vgg-master\test_vgg16.py
?????文件?????????845??2017-06-12?08:54??tensorflow-vgg-master\test_vgg19.py
?????文件????????1397??2017-06-12?08:54??tensorflow-vgg-master\test_vgg19_trainable.py
?????文件????????1921??2017-06-12?08:54??tensorflow-vgg-master\utils.py
?????文件????????4414??2017-06-12?08:54??tensorflow-vgg-master\vgg16.py
?????文件????????4616??2017-06-12?08:54??tensorflow-vgg-master\vgg19.py
?????文件????????6685??2017-06-12?08:54??tensorflow-vgg-master\vgg19_trainable.py
評論
共有 條評論