瓷砖瑕疵检测与数据增强

    # linux
    data_root_path = "data/MagneticTile"  # 数据路径
    test_file_path = "data/test.txt"  # 测试文件路径
    train_file_path = "data/train.txt"  # 测试文件路径
    # windows
    # data_root_path = "D:\ MagneticTile"  # 数据路径
    # test_file_path = "data\ test.txt"  # 测试文件路径
    # train_file_path = "data\ train.txt"  # 测试文件路径
    
    name_dict = {"MT_Blowhole": 0, "MT_Break": 1, "MT_Crack": 2,
             "MT_Fray": 3, "MT_Free": 4, "MT_Uneven":5}  # 字典