该项目涵盖从训练到测试的流程，详细说明了数据载入、缺陷数据生成、测试步骤以及可视化方法。在训练部分，涉及数据位置、缺陷数据生成和训练选项。测试部分包括测试文件目录、方式及结果保存路径。可视化章节提供了运行命令和存储位置。其他说明包含文件解释、运行环境建议和优化技巧。
一;训练
1.数据载入：
- 完美图位置：
datasets/trainB - 颜色遮罩位置：
lbp/mask_file/color-defect-->不需要mask - 表面异物位置：
lbp/mask_file/impurity-defect-->对应的mask位置：lbp/mask_file/mask - 纹理缺陷位置：（用于风格迁移）：
lbp/mask_file/transfer-defect-before–>对应的mask位置：lbp/mask_file/mask
2.缺陷数据生成：
python train.py --cuda --datarequire
- 如果不需要做数据生成：
python train.py -
最低0.47元/天 解锁文章
扫一扫
1166
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
