数据集训练

[zhangshilong2000]

您好，想请问一下，该代码是否支持自制数据集的训练与测试？如果支持该如何进行训练配置呢？

[zengyh1900]

hi @zhangshilong2000
可以的。但是你自己的数据集，你可能需要自己重写下数据集的是实现。
https://github.com/open-mmlab/PowerPaint/tree/dev?tab=readme-ov-file#training 训练配置和脚本在这里

[venshow-w]

@zengyh1900 请问inpainting训练时，构建数据集的mask是单目标物体的mask还是多目标无论，如果是多目标物体，prompt怎么确定对应哪一个目标呢，是在prompt中加入目标物体的坐标框吗？