以下是针对 aistudio.baidu.com 网站,生成'数据集'相关的 Selenium 前端自动化代码:

from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

# 初始化浏览器
browser = webdriver.Chrome()

# 打开网站
browser.get('https://aistudio.baidu.com/aistudio/datasetoverview')

# 等待页面加载完成
wait = WebDriverWait(browser, 10)
wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'dataset-card')))

# 点击'创建数据集'按钮
create_button = browser.find_element_by_xpath('//div[@class='create-dataset']/button')
create_button.click()

# 等待弹窗出现
wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'create-dialog')))

# 填写数据集名称
name_input = browser.find_element_by_xpath('//div[@class='create-dialog']//input[@name='name']')
name_input.send_keys('My Dataset')

# 选择数据集类型
type_select = browser.find_element_by_xpath('//div[@class='create-dialog']//select[@name='type']')
type_select.send_keys(Keys.DOWN)
type_select.send_keys(Keys.RETURN)

# 上传数据集文件
file_input = browser.find_element_by_xpath('//div[@class='create-dialog']//input[@type='file']')
file_input.send_keys('/path/to/my/dataset/file')

# 点击'创建'按钮
submit_button = browser.find_element_by_xpath('//div[@class='create-dialog']/button[@type='submit']')
submit_button.click()

# 等待数据集创建完成
wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'dataset-card')))

# 关闭浏览器
browser.quit()

这段代码实现了以下步骤:

  1. 初始化浏览器,打开 aistudio.baidu.com 网站;
  2. 等待页面加载完成,并点击'创建数据集'按钮;
  3. 等待弹窗出现,并填写数据集名称、选择数据集类型、上传数据集文件;
  4. 点击'创建'按钮,等待数据集创建完成;
  5. 关闭浏览器。

你可以根据自己的需求,对这段代码进行修改和扩展。

AI Studio 数据集创建自动化:Selenium 代码示例

原文地址: https://www.cveoy.top/t/topic/n471 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录