當你拿起手機準備發朋友圈的時候,會不會在選好圖片後卻不知道要寫什麼文案?
又或者是看到一處美景時,想不出別的高級形容詞,隻會說:“woc,太好看了吧!”
問題不大!AI可以幫你搞定!就像下麵這樣:
AI根據這張圖片生成的古詩是:
蕾蕾海河海,嶽峰嶽麓蔓。
不萌枝上春,自結心中線。
怎麼樣?還不錯吧!
二、實現思路
我認為研究AI其實就是在研究我們人類,為什麼這麼說呢?舉個例子:
我們在看圖說話時,一般都是先看看圖片上有什麼內容,比如一張圖裏有山有水還有很多的植物,那我們對這張圖的第一印象就是一張風景圖,然後我們再根據圖片上某一個具體的景物做詳細的描述。
其實AI也是類似的!
要想讓機器根據一張圖片寫古詩,可以分為以下三步:
最簡單的就是圖像分類,將圖像分到某一個類別裏,比如上麵那張圖的類別就是lakeside即湖邊。
拿到圖片的類別之後,也就有了這張圖的關鍵字,但關鍵字往往很短,不足以寫出一首完整的古詩,因此我們需要找到更多的關鍵詞。
當關鍵詞足夠多時(即能構成古詩的上闋,如:昔年旅南服,始識王荊州。),我們就把關鍵詞組合成古詩的上闋,然後送入模型。
最後得到模型的輸出即古詩的下闋,上、下闋組合後就是一首完整的古詩,我們還可以對古詩進行評估,分析一下情感傾向。
三、調用三個模型就能快速寫詩!
1.首先將paddlehub更新到最新的版本:
In [1]
!pip install --upgrade paddlehub -i https://pypi.tuna.tsinghua.edu.cn/simple
2.安裝所需要的模型:
圖像分類:xception71_imagenet
文本生成:ernie_gen_couplet
文本生成:ernie_gen_poetry
情感分析:senta_bow
In [2]
!hub install xception71_imagenet\u003d\u003d1.0.0 #圖像分類
!hub install ernie_gen_couplet\u003d\u003d1.0.0 #文本生成(對聯),獲取近似詞
!hub install ernie_gen_poetry\u003d\u003d1.0.0 #生成古詩詞
!hub install senta_bow\u003d\u003d1.2.0 #情感傾向分析
[2021-10-10 18:10:21,571] [ INFO] - Successfully installed senta_bow-1.2.0
2021-10-10 18:10:21,920 - INFO - Lock 140389601126352 released on /home/aistudio/.paddlehub/tmp/senta_bow
3.獲取圖片的分類,即該圖片是什麼