和同学想做一个图片分类的小项目,需要在百度图片上抓取一百多个关键词对应的图片各500张,代码如下,参考网址如代码中所示,在原代码中加入了从txt文件中逐行读取关键词和将图片按类别保存在不同文件夹下的功能,但其中多线程部分暂未实现。GitHub上有更详细的说明和步骤,欢迎大家移步。
GitHub网址:https://github.com/vivianLL/baidupictures
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#__author__ = 'leilu'
#参考网址:http://lovenight.github.io/2015/11/15/Python-3-%E5%A4%9A%E7%BA%BF%E7%A8%8B%E4%B8%8B%E8%BD%BD%E7%99%BE%E5%BA%A6%E5%9B%BE%E7%89%87%E6%90%9C%E7%B4%A2%E7%BB%93%E6%9E%9C/
import json
import itertools
import urllib
import requests
import os
import re
import codecs
import sys
reload(sys)
sys.setdefaultencoding("utf-8")
str_table = {
'_z2C$q': ':',
'_z&e3B': '.',
'AzdH3F': '/'
}
char_table = {
'w': 'a',
'k': 'b',
'v': 'c',
'1': 'd',
'j': 'e',
'u': 'f',
'2': 'g',
'i': 'h',
't': 'i',
'3': 'j',
'h': 'k',
's': 'l',
'4': 'm',
'g': 'n',
'5': 'o',
'r': 'p',
'q': 'q',
'6': 'r',
'f': 's',
'p': 't',
'7': 'u',
'e': 'v',
'o': 'w',
'8': '1',
'd': '2',
'n': '3',
'9': '4',
'c': '5',
'm': '6',
'0': '7',
'b': '8',
'l': '9',
'a': '0'
}
# str 的translate方法需要用单个字符的十进制unicode编码作为key
# value 中的数字会被当成十进制unicode编码转换成字符
# 也可以直接用字符串作为value
char_table = {ord(key): ord(value) for key, value in char_table.items()}
# 解码图片URL
def decode(url):
# 先替换字符串
for key, value in str_table.items():
url = url.replace(key, value)
# 再替换剩下的字符
return url.translate(char_table)
# 生成网址列表
def buildUrls(word):
word = urllib.quote(word)
url = r"http://image.baidu.com/search/acjson?tn=resultjson_com&ipn=rj&ct=201326592&fp=result&queryWord={word}&cl=2&lm=-1&ie=utf-8&oe=utf-8&st=-1&ic=0&word={word}&face=0&istype=2nc=1&pn={pn}&rn=60"
urls = (url.format(word=word, pn=x) for x in itertools.count(start=0, step=60))
return urls
# 解析JSON获取图片URL
re_url = re.compile(r'"objURL":"(.*?)"')
def resolveImgUrl(html):
imgUrls = [decode(x) for x in re_url.findall(html)]
return imgUrls
def downImg(imgUrl, dirpath, imgName):
filename = os.path.join(dirpath, imgName)
try:
res = requests.get(imgUrl, timeout=15)
if str(res.status_code)[0] == "4":
print(str(res.status_code), ":" , imgUrl)
return False
except Exception as e:
print("抛出异常:", imgUrl)
print(e)
return False
with open(filename, "wb") as f:
f.write(res.content)
return True
def mkDir(dirName):
dirpath = os.path.join(sys.path[0], dirName)
if not os.path.exists(dirpath):
os.mkdir(dirpath)
return dirpath
def mkdir(path):
path = path.strip()
isExists = os.path.exists(os.path.join("F:\RenYongguo\dogs",path))
if not isExists:
print ("新建了一个文件夹!")
dirpath = os.path.join("F:\RenYongguo\dogs",path)
os.makedirs(dirpath)
print dirpath
return dirpath
if __name__ == '__main__':
print("下载结果保存在F:\RenYongguo\dogs文件夹中。")
# print("=" * 50)
# word = input("请输入你要下载的图片关键词:\n")
f = open('F:\RenYongguo\dogtest.txt', 'r')
for line in f:
word = line.strip().decode('gbk', 'utf-8')
print word
dirpath = mkdir(word)
word = str(word)
urls = buildUrls(word)
index = 0
for url in urls:
print("正在请求:", url)
html = requests.get(url, timeout=10).content.decode('utf-8')
imgUrls = resolveImgUrl(html)
if len(imgUrls) == 0: # 没有图片则结束
break
for url in imgUrls:
if downImg(url, dirpath, str(index) + ".jpg"):
index += 1
print("已下载 %s 张" % index)
if index > 500:
break
if index > 500:
break
f.close()