資源簡介
最經典的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
評論
共有 條評論