yolov5 可以自定义网络，去掉配置文件

yolov5使用配置文件形式，

各种参数，跨文件，读懂整个网络不是特别方便，

如果想自己设计backbone，也不很方便，

特地调研了pytorch代码版，方便自己修改网络。

https://github.com/sbisdog/my_det/blob/731c348955c981e6cb73276794d969b61f826a33/auto_learn/public/detection/models/yolov5.py

代码：

  

   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      yolov5 official code
     
    
   

    

     

    
    

     

      https://github.com/ultralytics/yolov5
     
    
   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import sys
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      BASE_DIR = os.path.dirname(
     
    
   

    

     

    
    

     

       os.path.dirname(os.path.dirname(os.path.dirname(
     
    
   

    

     

    
    

     

       os.path.abspath(__file__)))))
     
    
   

    

     

    
    

     

      sys.path.append(BASE_DIR)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from public.path import pretrained_models_path
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from public.detection.models.anchor import YOLOV3Anchors
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
  
 

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

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