PowerPaint
PowerPaint copied to clipboard
数据集训练
您好,想请问一下,该代码是否支持自制数据集的训练与测试?如果支持该如何进行训练配置呢?
hi @zhangshilong2000 可以的。但是你自己的数据集,你可能需要自己重写下数据集的是实现。 https://github.com/open-mmlab/PowerPaint/tree/dev?tab=readme-ov-file#training 训练配置和脚本在这里
@zengyh1900 请问inpainting训练时,构建数据集的mask是单目标物体的mask还是多目标无论,如果是多目标物体,prompt怎么确定对应哪一个目标呢,是在prompt中加入目标物体的坐标框吗?