使用%%writetemplate修改YOLOv5 common.py触发KeyError: 'w'求助

当使用%%writetemplate魔法命令替换包含DetectMultiBackend类的YOLOv5 common.py文件时，代码中elif jit分支的f-string占位符{w}会被模板格式化逻辑误解析为模板变量，从而触发KeyError: 'w'报错。

以下是几种可行的解决方法：

• 转义f-string的大括号：将代码中所有f-string内的单个大括号{}替换为双大括号{{}}。例如原代码：

  print(f"Loading {w}...")
  

  修改为：

  print(f"Loading {{w}}...")
  

  这样%%writetemplate解析时会将{{w}}还原为{w}，既保留f-string的原有功能，又避免被当作模板变量解析。

• 替换f-string为其他字符串定义方式：如果不想修改大括号，可以改用str.format()方法或字符串拼接。比如将：

  msg = f"Model {w} loaded successfully"
  

  改为：

  msg = "Model {} loaded successfully".format(w)
  

  或者：

  msg = "Model " + w + " loaded successfully"
  

• 绕过模板命令直接写入文件：在Jupyter环境中，可先将完整的common.py代码赋值给变量，再通过Python原生文件操作写入，避免使用%%writetemplate。示例：

  new_common_code = """
  # 包含DetectMultiBackend的完整common.py代码
  class DetectMultiBackend(nn.Module):
      def __init__(self, weights='yolov5s.pt', device=None, dnn=False, data=None, fp16=False):
          super().__init__()
          # ... 其他初始化代码 ...
          elif jit:
              # 保留原逻辑，无需转义大括号
              print(f"Loading {w}...")
              # ... 剩余分支代码 ...
  """
  with open("common.py", "w", encoding="utf-8") as f:
      f.write(new_common_code)
  

内容的提问来源于stack exchange，提问作者Leprechault