AssertionError: Python module from file .\defaults.py must have ‘cfg‘ attr导入配置文件报错

AssertionError: Python module from file .\defaults.py must have 'cfg' attr
maskrcnn训练脚本导入配置文件报错

解决方案

# 在 defaults.py 文件的末尾添加
cfg = _C

# 训练脚本train.py提供配置文件路径
def main():
    # 创建参数解析器
    parser = argparse.ArgumentParser(description="PyTorch 目标检测训练")
    # 添加参数，用于配置文件、本地等级和跳过测试的选项
    parser.add_argument(
        "--config-file",
        default=".\config\defaults.py",
        metavar="FILE",
        help="配置文件的路径",
        type=str,
        ...
    #其他代码省略