UnrealSynthPro虚幻合成数据生成器专业版

可永久使用
三个月内免费在线技术支持
三个月内免费版本升级
价格¥799
立即下载

UnrealSynthPro 虚幻合成数据生成器专业版利用虚幻引擎的实时渲染能力搭建逼真的三维场景,为 YOLO 等 AI 模型的训练提供自动生成的图像和标注数据。UnrealSynthPro 生成的合成数据可用于深度学习模型的训练和验证,可以极大地提高各种行业细分场景中目标识别任务的实施效率,例如:安全帽检测、交通标志检测、施工机械检测、车辆检测、行人检测、船舶检测等。

界面

专业版说明:
1.支持不同的3D场景更换
2.导入模型种类、数量不受限制
3.支持任意视角输出图片,截图数量最大提升为5000张

1、UnrealSynthPro 合成数据工具包内容

UnrealSynth 基于 UE5 虚幻引擎开发,目前支持 YOLO 系列模型合成数据的生成,当前版本号 V1.0,主要文件和目录的组织结构如下:

目录 内容
Engine/ 发布本程序的原始软件的编码和资源文件,其中包含构件此程序的二进制编码和一些存放在 content 文件中的原始资产等
UnrealSynthPro/Binaries/ 本程序兼容系统及其他的二进制文件
UnrealSynthPro/Content/ 本程序中所使用的所有资产文件已被烘焙成 pak 包
UnrealSynthPro.exe 运行程序
LICENSE.md 开发包许可协议文件

运行UnrealSynthpro的推荐配置为:

2、UnrealSynthPro 合成数据生成

以下是以 YOLO 模型为例,详细讲述如何使用 UnrealSynthPro 虚幻引擎数据生成器专业版来生成为 YOLO 模型生成训练的合成数据。

打开 UnrealSynthPro 虚幻引擎合成数据生成器专业版,选择好地图后,点击【确定】按钮,进入虚幻场景编辑页面,输入【模型类别】参数后就可以开始导入模型,点击【导入 GLB 模型】弹出文件选择框,任意选择GLB 文件导入,可以重复导入多个GLB文件(记得每次导入前修改模型类别),导入 GLB 文件后的场景如下:

image-20231020180416311

将 GLB 文件添加到场景后,接下来就可以配置 UnrealSynth专业版 合成数据生成参数,参数配置说明如下:

点击【确定】后会在本地目录中...\UnrealSynthPro\Windows\UnrealSynthPro\Content\UserData 自动生成两个文件夹以及一个 yaml 文件:images、labels、test.yaml 文件。

UnrealSynthPro\Windows\UnrealSynthPro\Content\UserData
    |- images
        |-train
            |- 0.png
            |- 1.png
            |- 2.png
            |- ...
         |-val
            |- 0.png
            |- 1.png
            |- 2.png
            |- ...
    |- labels
        |-train
            |- 0.txt
            |- 1.txt
            |- 2.txt
            |- ...
        |-val
            |- 0.txt
            |- 1.txt
            |- 2.txt
            |- ...
    |- synth.yaml

UnrealSynthPro 合成数据已生成,可以利用数据集训练 YOLO 模型,会在 images 下生成两个图像目录:train 和 val。

train 目录表示训练图像数据目录,val 表示验证图像数据目录。

例如 train 目录下的图像集合:

img

同样在 labels 标注目录下也会生成两个标注目录:train 和 val。

train 目录表示标注训练数据目录,val 表示标注验证数据目录。

生成的 labels 标注数据格式如下:

0 0.68724 0.458796 0.024479 0.039815
0 0.511719 0.504167 0.021354 0.034259
0 0.550781 0.596759 0.039062 0.04537
0 0.549219 0.368519 0.023438 0.044444
0 0.47526 0.504167 0.009896 0.030556
0 0.470313 0.69537 0.027083 0.035185
0 0.570052 0.499074 0.016146 0.040741
0 0.413542 0.344444 0.022917 0.037037
0 0.613802 0.562037 0.015104 0.027778
0 0.477344 0.569444 0.017188 0.016667

生成的 synth.yaml 数据格式如下:

path:
train: images
val: images
test:
names:
 0: SM_Apple_02
 1: SM_Pear_01
 2: SM_FruitsOnPlate

3、利用 UnrealSynthPro 合成数据训练 YOLOv8 模型

数据集生成后有三个办法可以进行模型训练:使用 python 脚本、使用命令行、使用在线服务。

第一种是使用 python 脚本,需首先安装 ultralytics 包,训练代码如下所示:

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.yaml')  # build a new model from YAML
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # build from YAML and transfer weights

# Train the model
results = model.train(data='synth.yaml', epochs=100, imgsz=640)

第二种是使用命令行,需安装 YOLO 命令行工具,训练代码如下:

# Build a new model from YAML and start training from scratch
yolo detect train data=coco128.yaml model=yolov8n.yaml epochs=100 imgsz=640

# Start training from a pretrained *.pt model
yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640

# Build a new model from YAML, transfer pretrained weights to it and start training
yolo detect train data=coco128.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 imgsz=640

第三种是使用ultralytics hub 或者其他在线训练工具。

UnrealSynthPro虚幻合成数据生成器专业版 - NSDT
联系客服