duckduck_searchライブラリを使って、画像のURLを取得する方法

元教師

こんにちは！データサイエンティストの青木和也（https://twitter.com/kaizen_oni）です！

今回の記事では、duckduck_searchを使って画像のURLを取得する方法を紹介します！

私は「Practical Deep Learning for Coders」でDeep Learningの復習をしているのですが、第1章の

「Getting Started」の序盤からduckduck_searchの実装の違いにつまづいてしまったので、当レッスンのコードをduckduck_searchの最新版(ver 4.4)に書き換えたコードを紹介していきたいと思います！

同じようなところで詰まっている方に届けば幸いです！

元コード
修正版のコード
まとめ

元コード

第1章の授業用Kaggle notebook「Is it a bird? Creating a model from your own data」の元コードは以下の通りです。

!pip install duckduckgo-search==1.1
from duckduckgo_search import ddg_images
from fastcore.all import *

def search_images(term, max_images=30):
    print(f"Searching for '{term}'")
    return L(ddg_images(term, max_results=max_images)).itemgot('image')

ここでは、duckduckgo-searchのバージョンを1.1に指定してし、search_images関数の中でduckduck_searchのddg_images関数を使用しています。

このコードは以下のコードによって実行されるのですが、どうにもうまくいきません。

#NB: `search_images` depends on duckduckgo.com, which doesn't always return correct responses.
#    If you get a JSON error, just try running it again (it may take a couple of tries).
urls = search_images('bird photos', max_images=1)
urls[0]

修正版のコード

そこで元コードを以下のように書き換えます。

!pip install duckduck-search==4.4
from duckduckgo_search import DDGS
from fastcore.all import *

def search_images(term, max_images=30):
    print(f"Searching for '{term}'")
    with DDGS() as ddgs:
        images = list(ddgs.images(term,max_results = max_images))
        urls = []
        for image in images:
            print(image)
            urls.append(image['image'])
        return urls

上記のコードでは、以下の4つの手順を変更しています。