[09-24] 이미지 크롤링, 사진-그림 변환

이미지 크롤링 코드 (Sketch용)

from bs4 import BeautifulSoup # bs4, selenium 이용하여 이미지 크롤링 진행
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
import urllib.request
import time
import os

# 스크
def scroll_down():
    while True:
        time.sleep(1)
        # 페이지 맨 아래로 스크롤
        driver.find_element(By.XPATH, '//body').send_keys(Keys.END)
        time.sleep(1)
        try:
            # 더보기
            load_more_button = driver.find_element(By.XPATH, '//*[@id="islmp"]/div/div/div/div/div[1]/div[2]/div[2]/input')
            if load_more_button.is_displayed():
                load_more_button.click()
        except:
            pass
        time.sleep(1)
        try:
            no_more_content = driver.find_element(By.XPATH, '//div[@class="K25wae"]//*[text()="현풍읍 대구광역시 달성군"]')
            if no_more_content.is_displayed(): # 모든 이미지 확인을 표현
                break
        except:
            pass

if __name__ == "__main__":
    query = "fear sketch "
	  image_cnt = 50            # 사용에 따라 자율적으로 변화 가능 

    save_dir = "saved_image"  # 저장할 디렉토리 이름
    os.makedirs(save_dir, exist_ok=True)  # 디렉토리 생성 
    os.chdir(save_dir)  # 작업 디렉토리 변경

    driver = webdriver.Chrome()  # Chrome 웹 드라이버 실행
    URL = 'https://www.google.com/search?tbm=isch&q='
    driver.get(URL + query)  # 검색어를 포함한 URL로 이동

    scroll_down()  # 함수 호출

    # 이미지 정보 추출
    soup = BeautifulSoup(driver.page_source, 'html.parser')
    image_info_list = soup.find_all('img', class_='rg_i')
    image_and_name_list = []

    print('이미지 수집 시작')

    downlaod_cnt = 0
    for i in range(len(image_info_list)):
        if i == image_cnt:
            break
        if 'data-src' in image_info_list[i].attrs:
            save_image = image_info_list[i]['data-src']

            image_path = os.path.join(query.replace(' ', '_') + '_' + str(downlaod_cnt) + '.jpg')
            image_and_name_list.append((save_image, image_path))
            downlaod_cnt += 1

    # 다운로드
    for i in range(len(image_and_name_list)):
        urllib.request.urlretrieve(image_and_name_list[i][0], image_and_name_list[i][1])

    print('이미지 수집 종료')
    driver.close()  # 브라우저 닫기

Image-to-Image Conversion (ArtLine)