{"id":1628,"date":"2025-01-24T13:29:00","date_gmt":"2025-01-24T05:29:00","guid":{"rendered":"https:\/\/www.forillusion.com\/?p=1628"},"modified":"2025-02-14T11:39:02","modified_gmt":"2025-02-14T03:39:02","slug":"3-5-fashion-mnist","status":"publish","type":"post","link":"https:\/\/www.forillusion.com\/index.php\/3-5-fashion-mnist\/","title":{"rendered":"3.5 \u56fe\u50cf\u5206\u7c7b\u6570\u636e\u96c6\uff08Fashion-MNIST\uff09"},"content":{"rendered":"\n<p><div class=\"has-toc have-toc\"><\/div><\/p>\n\n\n\n<h2 class=\"wp-block-heading\">\u83b7\u53d6\u6570\u636e\u96c6<\/h2>\n\n\n\n<p>\u901a\u8fc7torchvision\u7684`torchvision.datasets`\u6765\u4e0b\u8f7d\u8fd9\u4e2a\u6570\u636e\u96c6\u3002\u7b2c\u4e00\u6b21\u8c03\u7528\u65f6\u4f1a\u81ea\u52a8\u4ece\u7f51\u4e0a\u83b7\u53d6\u6570\u636e\u3002\u901a\u8fc7\u53c2\u6570`train`\u6765\u6307\u5b9a\u83b7\u53d6\u8bad\u7ec3\u6570\u636e\u96c6\u6216\u6d4b\u8bd5\u6570\u636e\u96c6\u3002\u53c2\u6570`transform = transforms.ToTensor()`\u4f7f\u6240\u6709\u6570\u636e\u8f6c\u6362\u4e3a`Tensor`\uff0c\u5982\u679c\u4e0d\u8fdb\u884c\u8f6c\u6362\u5219\u8fd4\u56de\u7684\u662fPIL\u56fe\u7247\u3002`transforms.ToTensor()`\u5c06\u5c3a\u5bf8\u4e3a (H x W x C) \u4e14\u6570\u636e\u4f4d\u4e8e[0, 255]\u7684PIL\u56fe\u7247\u6216\u8005\u6570\u636e\u7c7b\u578b\u4e3a`np.uint8`\u7684NumPy\u6570\u7ec4\u8f6c\u6362\u4e3a\u5c3a\u5bf8\u4e3a(C x H x W)\u4e14\u6570\u636e\u7c7b\u578b\u4e3a`torch.float32`\u4e14\u4f4d\u4e8e[0.0, 1.0]\u7684`Tensor`\u3002<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code class=\"python\">mnist_train = torchvision.datasets.FashionMNIST(root='~\/Datasets\/FashionMNIST', train=True, download=True, transform=transforms.ToTensor())\nmnist_test = torchvision.datasets.FashionMNIST(root='~\/Datasets\/FashionMNIST', train=False, download=True, transform=transforms.ToTensor())\n\nprint(len(mnist_train), len(mnist_test)) #\u83b7\u53d6\u8be5\u6570\u636e\u96c6\u7684\u5927\u5c0f\n#output: 60000 10000\n\nfeature, label = mnist_train&#91;0]  # \u53ef\u4ee5\u901a\u8fc7\u4e0b\u6807\u6765\u8bbf\u95ee\u4efb\u610f\u4e00\u4e2a\u6837\u672c\nprint(feature.shape, label)  # Channel x Height x Width\n#output: torch.Size(&#91;1, 28, 28]) tensor(9)<\/code><\/pre>\n\n\n\n<h2 class=\"wp-block-heading\">\u8bfb\u53d6\u5c0f\u6279\u91cf<\/h2>\n\n\n\n<p>\u6570\u636e\u8bfb\u53d6\u7ecf\u5e38\u662f\u8bad\u7ec3\u7684\u6027\u80fd\u74f6\u9888\uff0c\u7279\u522b\u5f53\u6a21\u578b\u8f83\u7b80\u5355\u6216\u8005\u8ba1\u7b97\u786c\u4ef6\u6027\u80fd\u8f83\u9ad8\u65f6\u3002PyTorch\u7684`DataLoader`\u4e2d\u4e00\u4e2a\u5f88\u65b9\u4fbf\u7684\u529f\u80fd\u662f\u5141\u8bb8\u4f7f\u7528\u591a\u8fdb\u7a0b\u6765\u52a0\u901f\u6570\u636e\u8bfb\u53d6\u3002<\/p>\n\n\n\n<pre class=\"wp-block-code\"><code class=\"python\">batch_size = 256\nif sys.platform.startswith('win'):\n    num_workers = 0  # 0\u8868\u793a\u4e0d\u7528\u989d\u5916\u7684\u8fdb\u7a0b\u6765\u52a0\u901f\u8bfb\u53d6\u6570\u636e\nelse:\n    num_workers = 4\ntrain_iter = torch.utils.data.DataLoader(mnist_train, batch_size=batch_size, shuffle=True, num_workers=num_workers)\ntest_iter = torch.utils.data.DataLoader(mnist_test, batch_size=batch_size, shuffle=False, num_workers=num_workers)<\/code><\/pre>\n","protected":false},"excerpt":{"rendered":"<p>\u83b7\u53d6\u6570\u636e\u96c6 \u901a\u8fc7torchvision\u7684`torchvision.datasets`\u6765\u4e0b\u8f7d\u8fd9\u4e2a\u6570\u636e\u96c6\u3002\u7b2c\u4e00\u6b21\u8c03\u7528\u65f6\u4f1a\u81ea\u52a8\u4ece\u7f51\u4e0a\u83b7\u53d6 &#8230;<\/p>","protected":false},"author":1,"featured_media":1627,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[46,3],"tags":[45,44,12,22],"class_list":["post-1628","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-46","category-3","tag-45","tag-44","tag-12","tag-22"],"_links":{"self":[{"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/posts\/1628","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/comments?post=1628"}],"version-history":[{"count":1,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/posts\/1628\/revisions"}],"predecessor-version":[{"id":1717,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/posts\/1628\/revisions\/1717"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/media\/1627"}],"wp:attachment":[{"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/media?parent=1628"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/categories?post=1628"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.forillusion.com\/index.php\/wp-json\/wp\/v2\/tags?post=1628"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}