用圖像識(shí)別做爬蟲居然這么爽——下篇

歡迎關(guān)注哈希大數(shù)據(jù)微信公眾號(hào)【哈希大數(shù)據(jù)】


貓眼電影票房榜單中各電影實(shí)時(shí)票房數(shù)據(jù)和總票房數(shù)據(jù)是經(jīng)過加密的,咱們直接獲取不到,現(xiàn)在咱們就嘗試采用圖像識(shí)別的方法把實(shí)時(shí)票房數(shù)據(jù)和總票房數(shù)據(jù)爬下來缚柏,附帶把電影主演、上映時(shí)間叉信、宣傳海報(bào)鏈接等信息也獲取下來米辐。

本次爬取主要流程如下:

  1. 分析網(wǎng)頁結(jié)構(gòu)
  2. 用selenium調(diào)用chromedriver去驅(qū)動(dòng)chrome瀏覽器獲取網(wǎng)頁信息
  3. 將票房信息保存為圖片,利用PIL對(duì)圖片進(jìn)行二值化务冕、銳化等預(yù)處理
  4. 利用tesseract對(duì)保存的圖片進(jìn)行識(shí)別血当,得到票房數(shù)據(jù)
  5. 保存數(shù)據(jù)

接下來進(jìn)行詳細(xì)說明:

1. 分析網(wǎng)頁結(jié)構(gòu)

貓眼電影國內(nèi)票房榜URL為:https://maoyan.com/board/1,其結(jié)構(gòu)較為簡答禀忆,只展示國內(nèi)上線電影實(shí)時(shí)票房的前十名歹颓。

image

可爬取的字段也只有7個(gè):目前排名、實(shí)時(shí)票房數(shù)據(jù)油湖、總票房數(shù)據(jù)巍扛、電影名稱、電影主演乏德、上映時(shí)間撤奸、宣傳海報(bào)鏈接,不過我們要多記錄一個(gè)字段:當(dāng)前時(shí)間喊括,唯一好玩的點(diǎn)在于實(shí)時(shí)票房是加密的胧瓜,可以讓咱們?cè)囋嚹懿荒芾脠D像識(shí)別的方式把它搞定。

2. 獲取非加密數(shù)據(jù)

這里使用的是selenium模擬的方式獲取網(wǎng)頁信息郑什,之前也介紹過府喳,但沒用它直接解析數(shù)據(jù),前面一般是先把網(wǎng)頁信息獲取下來蘑拯,再用BeautifulSoup或PyQuery這兩個(gè)解析庫處理解析信息钝满,今天咱們看看如何用其API直接獲取數(shù)據(jù):
首先得到所有內(nèi)容標(biāo)簽:

contents_list = driver.find_elements_by_css_selector('dl.board-wrapper dd')

這里注意咱們用的是find_elements_by_css_selector,element后面加了s申窘。如果不加s弯蚜,find_element方法查找就只能得到第一個(gè)節(jié)點(diǎn),加了s后可獲得滿足條件的所有節(jié)點(diǎn)剃法,以列表類型返回碎捺。
對(duì)于標(biāo)簽的文本內(nèi)容的獲取也是非常簡單,例如提取電影排名:
contents['index']= content.find_element_by_css_selector('i.board-index').text
找到標(biāo)簽后加.text便可獲得標(biāo)簽內(nèi)容贷洲。獲取標(biāo)簽內(nèi)屬性值也很簡單,例如提取電影名稱: contents['title']= content.find_element_by_css_selector('img.board-img').get_attribute('alt')
找到標(biāo)簽后加.get_attribute('屬性名稱')便可獲得標(biāo)簽內(nèi)屬性值收厨。具體實(shí)例如下:

for content in contents_list:<br> &nbsp;&nbsp;&nbsp; contents['index'] = content.find_element_by_css_selector('i.board-index').text<br>; contents['image_url'] = content.find_element_by_css_selector('img.boardimg').get_attribute('src')<br>;contents['title'] = content.find_element_by_css_selector('img.boardimg').get_attribute('alt')<br> &nbsp;&nbsp;&nbsp; try:<br> contents['actor'] = content.find_element_by_css_selector('p.star').text<br>; except NoSuchElementException:<br>;contents['actor'] = ''<br>; contents['time'] = content.find_element_by_css_selector('p.releasetime').t;

3. 將票房信息保存為圖片,利用PIL對(duì)圖片進(jìn)行二值化优构、銳化等預(yù)處理操作
將想要獲取信息的標(biāo)簽保存圖片的一般步驟如下:
a) 截取當(dāng)前窗口(selenium不能直接截取單個(gè)標(biāo)簽)
b) 獲取標(biāo)簽位置坐標(biāo)和標(biāo)簽大小
c) 利用PIL包的crop函數(shù)根據(jù)標(biāo)簽的坐標(biāo)和大小將標(biāo)簽剪切保存下來

舉例如下:

image

上圖就是網(wǎng)頁截圖诵叁,利用save_screenshot函數(shù)可以獲得當(dāng)前窗口截圖,咱們需要將藍(lán)色方框內(nèi)的票房數(shù)據(jù)剪切保存下來俩块,所以我們需要它們的坐標(biāo)和大小黎休,幸運(yùn)的是咱們可以利用.location和.size函數(shù)得到標(biāo)簽的坐標(biāo)和大小浓领,最后再利用PIL包的crop函數(shù)根據(jù)標(biāo)簽的坐標(biāo)和大小將標(biāo)簽剪切保存下來。

以上步驟在截取信息在第一屏的情況很好用势腮,但在窗口超過了一屏联贩,且需要截取的元素在后面時(shí),就只能另辟蹊徑了捎拯。
咱們抓取貓眼電影的電影票房網(wǎng)頁就比較長泪幌,一屏根本展示不完。當(dāng)然咱們可以利用js下拉網(wǎng)頁然后截圖署照,這樣也能將下面網(wǎng)頁的信息截取到祸泪,但獲取到的標(biāo)簽位置坐標(biāo)的基準(zhǔn)點(diǎn)在網(wǎng)頁的左上角(如上圖提示),crop函數(shù)在裁剪時(shí)以當(dāng)前圖片左上角為基準(zhǔn)建芙,所以基準(zhǔn)不同截取出來的內(nèi)容肯定不對(duì)没隘。
在網(wǎng)上查到WebDriver.PhantomJS自帶的方法支持對(duì)整個(gè)網(wǎng)頁截屏,這不是太爽了禁荸,但當(dāng)我用的時(shí)候又懵逼了右蒲,PhantomJS是可以截取整個(gè)網(wǎng)頁,但截取結(jié)果是這樣的:

image

尼瑪要啥不給啥呀赶熟,同時(shí)程序也提示警告selenium已經(jīng)不再允許使用PhantomJS了瑰妄,怎么辦咱們還是默默回去用chrome吧!這時(shí)突然靈光一閃映砖,剛才不能用chrome的原因是location函數(shù)獲取的標(biāo)簽坐標(biāo)和crop函數(shù)在裁剪坐標(biāo)基準(zhǔn)不同间坐,咱們把它調(diào)一致不就可以了(嘖嘖嘖此處應(yīng)有掌聲),說干就干邑退,這個(gè)網(wǎng)頁上一共有10部電影我只用下拉10次就OK了竹宋,所以這里咱們來個(gè)for循環(huán),window.scrollTo函數(shù)用了下拉網(wǎng)頁瓜饥,它有兩個(gè)參數(shù)第一個(gè)是參數(shù)代表橫向拖動(dòng)像素?cái)?shù)逝撬,第二個(gè)代表縱向拖動(dòng)像素?cái)?shù)浴骂∨彝粒可是每次下拉多少呢?270px(像素)溯警,
咦你咋知道趣苏?
來來來告訴大家一個(gè)小技巧咱們用微信截圖是它會(huì)顯示截圖圖片大小,對(duì)本實(shí)例咱們每次下拉兩個(gè)電影票房之間的長度就好梯轻,如下圖其長度為270px食磕。

image

這樣應(yīng)該是能得到每個(gè)票房數(shù)據(jù)標(biāo)簽了,在得到票房數(shù)據(jù)標(biāo)簽后咱們還要進(jìn)行一下圖片預(yù)處理(二值化喳挑,銳化)彬伦,最后將標(biāo)簽保存下來滔悉,部分代碼如下:

length = n*270driver.execute_script('window.scrollTo(0, {}*270)'.format(n))# 保存當(dāng)前窗口為圖片driver.save_screenshot('b.png')# time.sleep(2)# 定位div地址location = div.location# print(location)# 得到div尺寸size = div.size# print(size)left = location['x']top = location['y'] - lengthright = location['x'] + size['width']bottom = location['y'] + size['height'] - length

4. 利用tesseract對(duì)保存的圖片進(jìn)行識(shí)別,得到票房數(shù)據(jù)
原始票房數(shù)據(jù)圖片:

image

預(yù)處理后的票房數(shù)據(jù)圖片:

image

對(duì)處理后的圖片進(jìn)行利用tesseract進(jìn)行識(shí)別處理单绑,部分代碼如下:

# 調(diào)用tesseract進(jìn)行文字?jǐn)?shù)字識(shí)別回官,目前對(duì)于淺顏色的字體識(shí)別率較低,若想提高識(shí)別率需要對(duì)圖片做預(yù)處理os.system('echo off')#關(guān)閉命令行窗口運(yùn)行命令的顯示os.system('tesseract' + ' ' + filename + ' ' + output + ' ' + '-l num+chi_tra') #默認(rèn)已配置好系統(tǒng)變量time.sleep(2)f = open(output + ".txt", encoding='utf-8')try:    t = f.readlines()[0]

目前在沒有去訓(xùn)練自己的數(shù)據(jù)識(shí)別包搂橙,直接利用tesseract的現(xiàn)有數(shù)據(jù)包歉提,通過多次試驗(yàn)對(duì)實(shí)時(shí)票房的識(shí)別率已達(dá)到97%以上,由于總票房字體顏色太淺区转,且字體較小目前識(shí)別率還比較低在60%左右苔巨。得到的數(shù)據(jù)具體如下圖所示:

image

本次初次嘗試?yán)脠D像識(shí)別解決爬蟲與到的問題,感覺確實(shí)比較好玩废离,目前獲取標(biāo)簽圖片的流程已經(jīng)打通侄泽,但圖像預(yù)處理和圖像識(shí)別部分還有大量工作要做,目前是采用第三方軟件Tesseract-OCR進(jìn)行的識(shí)別蜻韭,后面一方面會(huì)繼續(xù)研究Tesseract-OCR蔬顾,另一方面也會(huì)嘗試?yán)蒙疃葘W(xué)習(xí)的方法提高圖片文字的識(shí)別精度

image
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末湘捎,一起剝皮案震驚了整個(gè)濱河市诀豁,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌窥妇,老刑警劉巖舷胜,帶你破解...
    沈念sama閱讀 218,755評(píng)論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異活翩,居然都是意外死亡烹骨,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,305評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門材泄,熙熙樓的掌柜王于貴愁眉苦臉地迎上來沮焕,“玉大人,你說我怎么就攤上這事拉宗÷褪鳎” “怎么了?”我有些...
    開封第一講書人閱讀 165,138評(píng)論 0 355
  • 文/不壞的土叔 我叫張陵旦事,是天一觀的道長魁巩。 經(jīng)常有香客問我,道長姐浮,這世上最難降的妖魔是什么谷遂? 我笑而不...
    開封第一講書人閱讀 58,791評(píng)論 1 295
  • 正文 為了忘掉前任借笙,我火速辦了婚禮粘昨,結(jié)果婚禮上弹沽,老公的妹妹穿的比我還像新娘脆荷。我一直安慰自己,他們只是感情好集晚,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,794評(píng)論 6 392
  • 文/花漫 我一把揭開白布掠廓。 她就那樣靜靜地躺著,像睡著了一般甩恼。 火紅的嫁衣襯著肌膚如雪蟀瞧。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,631評(píng)論 1 305
  • 那天条摸,我揣著相機(jī)與錄音悦污,去河邊找鬼。 笑死钉蒲,一個(gè)胖子當(dāng)著我的面吹牛切端,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播顷啼,決...
    沈念sama閱讀 40,362評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼踏枣,長吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來了钙蒙?” 一聲冷哼從身側(cè)響起茵瀑,我...
    開封第一講書人閱讀 39,264評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎躬厌,沒想到半個(gè)月后马昨,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,724評(píng)論 1 315
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡扛施,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,900評(píng)論 3 336
  • 正文 我和宋清朗相戀三年鸿捧,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片疙渣。...
    茶點(diǎn)故事閱讀 40,040評(píng)論 1 350
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡匙奴,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出妄荔,到底是詐尸還是另有隱情泼菌,我是刑警寧澤,帶...
    沈念sama閱讀 35,742評(píng)論 5 346
  • 正文 年R本政府宣布懦冰,位于F島的核電站灶轰,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏刷钢。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,364評(píng)論 3 330
  • 文/蒙蒙 一乳附、第九天 我趴在偏房一處隱蔽的房頂上張望内地。 院中可真熱鬧伴澄,春花似錦、人聲如沸阱缓。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,944評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽荆针。三九已至敞嗡,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間航背,已是汗流浹背喉悴。 一陣腳步聲響...
    開封第一講書人閱讀 33,060評(píng)論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留玖媚,地道東北人箕肃。 一個(gè)月前我還...
    沈念sama閱讀 48,247評(píng)論 3 371
  • 正文 我出身青樓,卻偏偏與公主長得像今魔,于是被迫代替她去往敵國和親勺像。 傳聞我的和親對(duì)象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,979評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容