AI-Studio预设模型训练实操--数据集制作

一、注册，并取得算力

不多赘述

二、选择模型

这里选择了目标检测模型

RT-DETR_HGNetV2_h

点击运行一下创造项目

建议选择消耗最大的卡，每个月会有时间限制

这里已经打开一个了，所以选不了

三、数据集

如果只是熟悉操作的话可以选择预设已有的数据集

点击启动项目上方的修改按钮

在这里添加数据集

3.1预设数据集只要点击添加就可以了，但创建数据集要考虑的事就很多了

这里使用labelme进行数据集创建

 先下载pip

windows上pip安装及使用详解-CSDN博客

然后通过指令下载labelme：

pip install labelme

速度慢的话可以在后面跟上国内的pip源，如：

pip install labelme -i https://pypi.tuna.tsinghua.edu.cn/simple

然后输入labelme打开软件，要是打不开就试试管理员终端

处理图片信息

大多数模型训练不需要过于清晰的图片，所以这里要降低图片的像素，具体降低多少根据实际情况而定

import cv2
import os
#批量重新设置图片尺寸大小
path = 'A:\\opencv\\'
for filename in os.listdir(path):
    file = path + filename
    img = cv2.imread(file)
    img2 = cv2.resize(img, (0, 0)