引入
??熱狗資料集包含2800包含熱狗和其他食物的影像，下載地址如下：
??原始：https://apache-mxnet.s3-accelerate.amazonaws.com/gluon/dataset/hotdog.zip
??微云：鏈接：https://share.weiyun.com/lTrk8xq4 密碼：fks7u4
1 影像載入
from torchvision.datasets import ImageFolder
def load_image(save_home="D:/Data/Image/hotdog/"):
"""
影像載入
"""
# 訓練集和測驗集
return ImageFolder(save_home + "train"), ImageFolder(save_home + "test")
def test1():
"""測驗之影像載入"""
image_tr, image_te = load_image()
if __name__ == '__main__':
test1()
2 影像展示
def test2():
"""測驗之影像展示"""
def show_images(imgs, num_rows, num_cols, scale=1.4):
import matplotlib.pyplot as plt
_, axes = plt.subplots(num_rows, num_cols, figsize=(num_cols * scale, num_rows * scale))
for i in range(num_rows):
for j in range(num_cols):
axes[i][j].imshow(imgs[i * num_cols + j])
axes[i][j].axes.get_xaxis().set_visible(False)
axes[i][j].axes.get_yaxis().set_visible(False)
plt.show()
image_tr, image_te = load_image()
hotdogs = [image_tr[i][0] for i in range(8)]
not_hotdogs = [image_tr[-i - 1][0] for i in range(8)]
show_images(hotdogs + not_hotdogs, 2, 8)
if __name__ == '__main__':
test2()
??輸出如下：
致謝
??感謝李沐、Aston Zhang等老師的這本《動手學深度學習》一書，為鄙人學習深度學習提供了很大的幫助，本文一系列關于深度學習的博客均無侵權之意，只為記錄自己的深度學習歷程，
??專案Github地址：https://github.com/ShusenTang/Dive-into-DL-PyTorch
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/289408.html
標籤：其他