半自动下载removebg图片（多进程版）

原创

已于 2024-05-16 15:35:16 修改

· 311 阅读

0 ·

版权

文章标签：

#selenium #removebg

于 2024-05-16 15:33:40 首次发布

之前写好了一篇单进程版的：半自动下载removebg图片-优快云博客

会嫌弃比较慢的话可以改成多进程代码，如下：

import os
from multiprocessing import Pool, Manager
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.common.exceptions import NoSuchElementException
from time import sleep

BASE_URL = 'https://www.remove.bg/zh/upload'

#保存已经完成过的urls
with open('runs.txt','r') as fp:
    runs_ls = fp.readlines()
    runs_ls = [i.strip() for i in runs_ls]

def exe_generate(bro):
    sleep(10)
    bro.find_element_by_xpath("//*[@id='page-content']/div[2]/div/div/div/div[2]/div[2]/div[2]/button").click()

def initialize_browser():
    profile_directory = f"./profiles/{os.getpid()}"
    os.makedirs(profile_directory, exist_ok=Tru