使用coremltools转换PyTorch模型至Core ML时遇pad算子未实现错误求助

这个错误不是你的代码逻辑问题，是当前使用的coremltools版本未实现对PyTorch中特定pad算子的转换支持。以下是可行的解决方向：

• 升级coremltools到最新版本
  先查看当前版本：

  pip show coremltools
  

  若版本较旧，执行升级命令：

  pip install --upgrade coremltools
  

  新版本通常会补充更多算子的支持，大概率能解决这个问题。

• 替换模型中的Pad实现
  如果升级后仍报错，说明模型里的Pad操作是coremltools暂不支持的类型（比如自定义Pad逻辑、特殊参数组合）。可以把模型中的Pad层替换为PyTorch标准的、coremltools支持的Pad类，比如nn.ZeroPad2d、nn.ReplicationPad2d、nn.ReflectionPad2d等。

• 自定义算子转换逻辑（进阶）
  若必须保留原Pad逻辑，可给coremltools注册自定义的pad算子转换函数。示例代码如下（需根据你模型中Pad的实际参数调整）：

  import coremltools as ct
  from coremltools.converters.mil import register_torch_op
  from coremltools.converters.mil.frontend.torch.ops import _get_inputs
  
  @register_torch_op(op_type="pad")
  def convert_pad(context, node):
      inputs = _get_inputs(context, node)
      x = inputs[0]
      pad = inputs[1].val.tolist()
      mode = node.attr.get("mode", "constant")
      value = inputs[2].val if len(inputs) > 2 else 0.0
      
      # 匹配coreml支持的pad模式
      if mode == "constant":
          pad_op = context.add_op("pad", x=x, pad=pad, mode="constant", constant_value=value, name=node.name)
      elif mode == "reflect":
          pad_op = context.add_op("pad", x=x, pad=pad, mode="reflect", name=node.name)
      elif mode == "replicate":
          pad_op = context.add_op("pad", x=x, pad=pad, mode="edge", name=node.name)
      else:
          raise ValueError(f"不支持的pad模式：{mode}")
      context.add(pad_op.outputs[0], node.name)
  

  注册后再执行转换流程即可。

• 切换TorchScript生成方式
  尝试用torch.jit.script代替torch.jit.trace来生成脚本模型，有时候能避免跟踪捕获到非标准操作：

  # 替换原来的traced_model生成代码
  scripted_model = torch.jit.script(jit_model_wrapper).to(device)
  model1 = ct.convert(
      scripted_model,
      source='pytorch',
      inputs=[ct.ImageType(name='image',shape=image.shape), ct.ImageType(name='mask',shape=mask.shape)]
  )
  

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