Pythonで特定URLの画像ファイルを取得する

概要

機械学習等を行う場合、多数の画像ファイルを取得する必要がある。
その場合、WEBサイトから画像ファイルを取得する必要がある為、
URLの画像ファイルをダウンロードする際に使用するサンプルPGを下記に記載。

前提条件

• requests
• python3.7

サンプルPG

get_img.py

import requests

# ファイル出力先
OUTPUT_PATH = "D:\\img"

def main():
    url = "https://example.com/sample/sample.jpg"
    get_img(url)


def get_img(url):

    # URLにGET通信でアクセス
    res = requests.get(url)

    # 200以外のステータスの場合は処理終了
    if res.status_code != 200:
        return

    # コンテンツタイプを確認
    content_type = res.headers["content-type"]

    # コンテンツタイプが画像出ない場合処理終了
    if 'image' not in content_type:
        return

    # 画像データを変数に代入
    image = res.content

    # URLからファイル名を取得(URLの末尾を取得)
    f_name = url.split("/")[-1]

    # 画像ファイルを出力
    with open(OUTPUT_PATH + "/" + f_name, "wb") as fout:
        fout.write(image)

if __name__ == "__main__":
    main()

参考サイト

• PythonでURLからファイル名を取得してみる