silu to ONNX opset version 12 is not supported

yolov7导出onnx时报错：

Starting ONNX export with onnx 1.12.0...
ONNX export failure: Exporting the operator silu to ONNX opset version 12 is not supported. Please open a bug to request ONNX export support for the missing operator.
 

方法1，把silu替换掉：

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import models
     
    
   

    

     

    
    

     

      from models.experimental import attempt_load
     
    
   

    

     

    
    

     

      from utils.activations import Hardswish, SiLU
     
    
   

    

     

    
    

     

      from utils.general import set_logging, check_img_size
     
    
   

    

     

    
    

     

      from utils.torch_utils import select_device
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def replace_module(module, replaced_module_type, new_module_type, replace_func=None) -> nn.Module:
     
    
   

    

     

    
    

     
    """
     
    
   

    

     

    
    

     

       Replace given type in module to a new type. mostly used in deploy.
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

       Args:
     
    
   

    

     

    
    

     

       module (nn.Module): model to apply replace operation.
     
    
   

    

     

    
    

     

       replaced_module_type (Type): module type to be replaced.
     
    
   

    

     

    
    

     

       new_module_type (Type)
     
    
   

    

     

    
    

     

       replace_func (function): python function to describe replace logic. Defalut value None.
     
    
   

    

     

    
    

     

      
     
    
   

    

     

    
    

     

       Returns:
     
    
   

    

     

    
    

     

       model (nn.Module): module that already been replaced.
     
    
   

    

     

    
    

     

       """
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    def default_replace_func(replaced_module_type, new_module_type):
     
    
   

    

     

    
    

     
        return new_module_type()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    if replace_func is None:
     
    
   

    

     

    
    

     

              replace_func = default_replace_func
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          model = module
     
    
   

    

     

    
    

     
    if isinstance(module, replaced_module_type):
     
    
   

    

     

    
    

     

              model = replace_func(replaced_module_type, new_module_type)
     
    
   

    

     

    
    

     
    else:  # recurrsively replace
     
    
   

    

     

    
    

     
        for name, child in module.named_children():
     
    
   

    

     

    
    

     

                  new_child = replace_module(child, replaced_module_type, new_module_type)
     
    
   

    

     

    
    

     
            if new_child is not child:  # child is already replaced
     
    
   

    

     

    
    

     

                      model.add_module(name, new_child)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    return model
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          model = replace_module(model, nn.SiLU, SiLU)
     
    
   

    

     

    
    

     

          model = replace_module(model, nn.Hardswish, Hardswish)
     
    
  
 

  
 
 

方法2：

把torch从原版本1.7.0升级到1.10.

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/126756687