{"id":2213,"date":"2018-12-26T12:30:43","date_gmt":"2018-12-26T04:30:43","guid":{"rendered":"https:\/\/yanjingang.com\/blog\/?p=2213"},"modified":"2020-06-28T10:59:52","modified_gmt":"2020-06-28T02:59:52","slug":"%e5%a6%82%e4%bd%95%e7%94%a8python%e8%a7%a3%e6%9e%90cifar10%e6%95%b0%e6%8d%ae%e9%9b%86%e5%9b%be%e7%89%87","status":"publish","type":"post","link":"https:\/\/yanjingang.com\/blog\/?p=2213","title":{"rendered":"\u5982\u4f55\u7528python\u89e3\u6790cifar10\u6570\u636e\u96c6\u56fe\u7247"},"content":{"rendered":"

\u6982\u8ff0<\/h1>\n

\u901a\u7528\u56fe\u50cf\u5206\u7c7b\u516c\u5f00\u7684\u6807\u51c6\u6570\u636e\u96c6\u5e38\u7528\u7684\u6709CIFAR<\/a>\u3001ImageNet<\/a>\u3001COCO<\/a>\u7b49\uff0c\u5e38\u7528\u7684\u7ec6\u7c92\u5ea6\u56fe\u50cf\u5206\u7c7b\u6570\u636e\u96c6\u5305\u62ecCUB-200-2011<\/a>\u3001Stanford Dog<\/a>\u3001Oxford-flowers<\/a>\u7b49\u3002\u5176\u4e2dImageNet\u6570\u636e\u96c6\u89c4\u6a21\u76f8\u5bf9\u8f83\u5927\uff0c\u5927\u91cf\u7814\u7a76\u6210\u679c\u57fa\u4e8eImageNet\u3002ImageNet\u6570\u636e\u4ece2010\u5e74\u6765\u7a0d\u6709\u53d8\u5316\uff0c\u5e38\u7528\u7684\u662fImageNet-2012\u6570\u636e\u96c6\uff0c\u8be5\u6570\u636e\u96c6\u5305\u542b1000\u4e2a\u7c7b\u522b\uff1a\u8bad\u7ec3\u96c6\u5305\u542b1,281,167\u5f20\u56fe\u7247\uff0c\u6bcf\u4e2a\u7c7b\u522b\u6570\u636e732\u81f31300\u5f20\u4e0d\u7b49\uff0c\u9a8c\u8bc1\u96c6\u5305\u542b50,000\u5f20\u56fe\u7247\uff0c\u5e73\u5747\u6bcf\u4e2a\u7c7b\u522b50\u5f20\u56fe\u7247\u3002<\/p>\n

\u7531\u4e8eImageNet\u6570\u636e\u96c6\u8f83\u5927\uff0c\u4e0b\u8f7d\u548c\u8bad\u7ec3\u8f83\u6162\uff0c\u4e3a\u4e86\u65b9\u4fbf\u5feb\u901f\u5b66\u4e60\u56fe\u50cf\u5206\u7c7b\uff0c\u6211\u4eec\u4f7f\u7528CIFAR10<\/a>\u6570\u636e\u96c6\u3002\u672c\u6587\u8bb2\u89e3\u5982\u4f55\u5c06\u6570\u636e\u96c6\u53cd\u5411\u91cd\u5efa\u4e3argb\u5f69\u8272\u56fe\u7247\u3002<\/p>\n

cifar10\u6570\u636e\u96c6\u5171\u670960000\u5f20\u5f69\u8272\u56fe\u50cf\uff0c\u56fe\u50cf\u5927\u5c0f\u662f3\u901a\u9053\u768432*32\uff0c\u5206\u4e3a10\u4e2a\u7c7b\uff0c\u6bcf\u7c7b6000\u5f20\u56fe\u3002\u8fd9\u91cc\u9762\u670950000\u5f20\u7528\u4e8e\u8bad\u7ec3\uff0c\u6784\u6210\u4e865\u4e2a\u8bad\u7ec3\u6279\uff0c\u6bcf\u4e00\u627910000\u5f20\u56fe\uff1b\u53e6\u591610000\u7528\u4e8e\u6d4b\u8bd5\uff0c\u5355\u72ec\u6784\u6210\u4e00\u6279\u3002\u6d4b\u8bd5\u6279\u7684\u6570\u636e\u91cc\uff0c\u53d6\u81ea10\u7c7b\u4e2d\u7684\u6bcf\u4e00\u7c7b\uff0c\u6bcf\u4e00\u7c7b\u968f\u673a\u53d61000\u5f20\u3002\u62bd\u5269\u4e0b\u7684\u5c31\u968f\u673a\u6392\u5217\u7ec4\u6210\u4e86\u8bad\u7ec3\u6279\u3002\u6ce8\u610f\u4e00\u4e2a\u8bad\u7ec3\u6279\u4e2d\u7684\u5404\u7c7b\u56fe\u50cf\u5e76\u4e0d\u4e00\u5b9a\u6570\u91cf\u76f8\u540c\uff0c\u603b\u7684\u6765\u770b\u8bad\u7ec3\u6279\uff0c\u6bcf\u4e00\u7c7b\u90fd\u67095000\u5f20\u56fe\u3002
\n\u6570\u636e\u7684\u4e0b\u8f7d\uff1a\u5171\u6709\u4e09\u4e2a\u7248\u672c\uff1a<\/p>\n

python\uff1a http:\/\/www.cs.toronto.edu\/~kriz\/cifar-10-python.tar.gz
\nMATLAB\uff1a http:\/\/www.cs.toronto.edu\/~kriz\/cifar-10-matlab.tar.gz
\nbin\uff1a http:\/\/www.cs.toronto.edu\/~kriz\/cifar-10-binary.tar.gz<\/p>\n

\u4e0b\u56fe\u662f\u4ece\u6bcf\u4e2a\u7c7b\u522b\u4e2d\u968f\u673a\u62bd\u53d6\u4e8610\u5f20\u56fe\u7247\uff0c\u5c55\u793a\u4e86\u6240\u6709\u7684\u7c7b\u522b\u3002<\/p>\n

\"\"<\/a><\/p>\n

<\/p>\n

\u89e3\u6790<\/h1>\n

\u6211\u4eec\u4e0b\u8f7dpython\u7248\u672c\u7684\u6570\u636e\u96c6\u00a0cifar-10-python.tar.gz<\/a>\u00a0\u5e76\u89e3\u538b\uff0c\u53ef\u4ee5\u770b\u5230\u67095\u4e2atrainbatch\u6587\u4ef6\uff0c1\u4e2atestbatch\u6587\u4ef6\uff1a<\/p>\n

\"\"<\/a><\/p>\n

\u6587\u4ef6\u4f7f\u7528pickle\u683c\u5f0f\u538b\u7f29\uff0c\u6240\u4ee5\u9700\u8981\u4f7f\u7528python\u7684pickle\u5e93\u8fdb\u884c\u8bfb\u53d6\uff1a<\/p>\n

import cPickle\r\ndata = {}\r\nwith open(file, 'rb') as fo:\r\n     data = cPickle.load(fo)\r\n     print(data.keys())\r\n     #data\u662f\u4e2adict\uff0c\u5305\u542b\u56fe\u7247\u50cf\u7d20\u6570\u636edata\u548c\u5bf9\u5e94label\u6570\u636e{'data':[],'labels':[],'...'}<\/code><\/pre>\n
\u5b8c\u6574\u7684\u8bfb\u53d6\u5e76\u751f\u6210train\/test\u56fe\u7247\u7684\u4ee3\u7801\uff1a<\/p>\n
#!\/usr\/bin\/env python\r\n# -*- coding: utf-8 -*-\r\n\r\n\"\"\"\r\nFile: read_cifar10.py\r\nDesc: \u8bfb\u53d6cifar10\u6570\u636e\u96c6\u7684\u56fe\u7247\u548clabel\uff0c\u5e76\u751f\u6210\u4ee5\u5e8f\u53f7+label\u547d\u540d\u7684\u56fe\u7247\r\nAuthor:yanjingang(yanjingang@mail.com)\r\nDate: 2018\/12\/25 23:12\r\nCmd: nohup python read_cifar10.py >log\/read_cifar10.log &\r\n\"\"\"\r\n\r\nimport os\r\nimport random\r\nimport platform\r\nimport numpy\r\nimport subprocess\r\nfrom PIL import Image\r\nimport utils\r\nimport cPickle\r\n\r\n# cifar10\u8bad\u7ec3\u96c6\u76ee\u5f55  from: http:\/\/www.cs.toronto.edu\/~kriz\/cifar-10-python.tar.gz\r\ncifar10_path = '.\/data\/cifar-10-batches-py\/'\r\ntrain_batchs = [\r\n    cifar10_path + 'data_batch_1',\r\n    cifar10_path + 'data_batch_2',\r\n    cifar10_path + 'data_batch_3',\r\n    cifar10_path + 'data_batch_4',\r\n    cifar10_path + 'data_batch_5'\r\n]\r\ntest_batchs = [cifar10_path + 'test_batch']\r\n\r\n# \u8bfb\u53d6\u51fa\u7684\u56fe\u7247\u5b58\u653e\u4f4d\u7f6e\r\noutput_path = '.\/data\/'\r\n\r\n# label\u542b\u4e49\r\nlabel_list = [\"airplane\", \"automobile\", \"bird\", \"cat\", \"deer\", \"dog\", \"frog\", \"horse\", \"ship\", \"truck\"]\r\n\r\n\r\ndef reader_cifar10(batchs, path='train'):\r\n    \"\"\"\u8bfb\u53d6cifar10\u6570\u636e\u96c6\u7684\u56fe\u7247\u548clabel\uff0c\u5e76\u751f\u6210\u4ee5\u5e8f\u53f7-label-labelname\u547d\u540d\u7684\u56fe\u7247\"\"\"\r\n    id = 0  # \u56fe\u7247\u96c6\u5e8f\u53f7\r\n    for file in batchs:\r\n        data = {}\r\n        with open(file, 'rb') as fo:\r\n            data = cPickle.load(fo)\r\n            print(data.keys())\r\n            # print(data['data'])\r\n        for i in xrange(len(data['data'])):  # \u904d\u5386\u56fe\u7247\u50cf\u7d20\u6570\u636e\r\n            id += 1\r\n            # \u8bfb\u53d6\u5355\u5f20\u56fe\u7247\u6570\u636e\r\n            img = data['data'][i]\r\n            label = data['labels'][i]\r\n            label_name = label_list[label]\r\n            print(img)\r\n            print(len(img))\r\n            print(str(label) + ' : ' + label_name)\r\n\r\n            # \u91cd\u5efargb\u5f69\u8272\u56fe\u7247\r\n            img = img.reshape(3, 32, 32)\r\n            print(img)\r\n            print(img.shape)\r\n            r = Image.fromarray(img[0]).convert('L')\r\n            g = Image.fromarray(img[1]).convert('L')\r\n            b = Image.fromarray(img[2]).convert('L')\r\n            new_img = Image.merge('RGB', (r, g, b))\r\n\r\n            # \u4fdd\u5b58\u56fe\u7247(\u5e8f\u53f7-label-labelname.png)\r\n            utils.mkdir(output_path + path)\r\n            save_file = output_path + path + '\/' + str(id) + '-' + str(label) + '-' + label_name + '.png'\r\n            new_img.save(save_file)\r\n            print save_file\r\n            # if id > 10:\r\n            #    break\r\n        # break\r\n\r\n\r\nif __name__ == '__main__':\r\n    # \u8bfb\u53d6\u8bad\u7ec3\u96c6\r\n    # reader_cifar10(train_batchs, path='train')\r\n    # \u8bfb\u53d6\u6d4b\u8bd5\u96c6\r\n    reader_cifar10(test_batchs, path='test')<\/code><\/pre>\n
 <\/p>\n
\u5355\u56fe\u7247\u65e5\u5fd7\uff1a<\/p>\n
\"\"<\/a><\/p>\n
*\u6ce8\uff1a\u53ef\u4ee5\u770b\u5230\u5355img\u50cf\u7d20\u6570\u636e\u662f\u4e00\u4e2a3072\u957f\u5ea6\u7684\u4e00\u7ef4\u6570\u7ec4\uff0c\u5148reshape\u62103*32*32\u7684r\/g\/b\u6570\u636e\u683c\u5f0f\u518dmerge\u6210\u4e00\u5f20\u5f69\u56fe\u3002<\/p>\n
 <\/p>\n
\u751f\u6210\u7684\u672c\u5730\u56fe\u7247\uff1a\u00a0\"\"<\/a><\/p>\n
yan 18.12.25 23:36<\/p>\n
 <\/p>\n
\u53c2\u8003\uff1ahttps:\/\/blog.csdn.net\/jinxiaonian11\/article\/details\/80192161<\/span><\/p>\n","protected":false},"excerpt":{"rendered":"
\u6982\u8ff0 \u901a\u7528\u56fe\u50cf\u5206\u7c7b\u516c\u5f00\u7684\u6807\u51c6\u6570\u636e\u96c6\u5e38\u7528\u7684\u6709CIFAR\u3001ImageNet\u3001COCO\u7b49\uff0c\u5e38\u7528\u7684\u7ec6\u7c92\u5ea6\u56fe\u50cf\u5206\u7c7b\u6570\u636e\u96c6 […]<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":[],"categories":[535],"tags":[709,536,710,711],"_links":{"self":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/2213"}],"collection":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=2213"}],"version-history":[{"count":0,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=\/wp\/v2\/posts\/2213\/revisions"}],"wp:attachment":[{"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=2213"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=2213"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/yanjingang.com\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=2213"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}