如何训练自己的数据集？以及如何获取您训练的pth？

[CRONAlDO7531]

您好，我想知道如何训练自己的工业检测数据集，这个好像需要文本数据集，所以想了解怎么构建数据集？预训练文件如何获取？

[hk-zh]

您好，您是想问如果训练您自己的数据集而不是公开数据集吗？如果您想在自己数据集上做训练，你的数据集需要有对于不同defect type的标注还有在图上对应的语意分割。对于每个defect type，您需要有一些文字描述。
比如："Product A has a hole on it.", "visible hole on Product A". 在我们论文中的appendix中有对于defect type描述的样例。
对于我们的checkpoint，根据内部政策不能公开。但是训练的源代码已经公开，可以根据readme.md中的描述来训练。