devbox
Li_阴宅
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

基于YOLOv5全系列模型【n/s/m/l/x】开发构建商品商标检测识别系统，融合SegmentAnything（SAM）模型实现自动化商标区域分割抠图_yolov5smlx

作者：Li_阴宅 | 2024-07-14 20:06:42

赞

踩

yolov5smlx

目标检测系列的项目我们有过非常多的项目实践，图像分割也是如此，但是将目标检测产生的结果作为prompt整合输入CV大模型SAM中来实现自动化的ROI区域分割，然后将对应区域抠图存储下来这个倒是没有做过，本文的主要目的就是想要建立：YOLO+SAM的自动化目标检测抠图流程，首先看下实例效果：

简单看下目标检测数据集：

YOLOv5是一种快速、准确的目标检测模型，由Glen Darby于2020年提出。相较于前两代模型，YOLOv5集成了众多的tricks达到了性能的SOTA：
技术原理：
YOLOv5同样采用单个神经网络模型来完成目标检测任务，但采用了新的神经网络架构，融合了领先的轻量级模型设计理念。YOLOv5使用较小的骨干网络和新的检测头设计，以实现更快的推断速度，并在不降低精度的前提下提高目标检测的准确性。
亮点：
YOLOv5在模型结构上进行了改进，引入了更先进的轻量级网络架构，因此在速度和精度上都有所提升。
YOLOv5支持更灵活的模型大小和预训练选项，可以根据任务需求选择不同大小的模型，同时提供丰富的数据增强扩展、模型集成等方法来提高检测精度。YOLOv5通过使用更简洁的代码实现，提高了模型的易用性和可扩展性。

训练数据配置文件如下：

  1. # Dataset
  2. path: ./dataset
  3. train:
  4.   - images/train
  5. val:
  6.   - images/test
  7. test:
  8.   - images/test
  9.  
  10.  
  11. # number of classes
  12. nc: 15  
  13.  
  14.  
  15.  
  16. # class names
  17. names: ['361', 'Adidas', 'Anta', 'CocaCola', 'Erke', 'Heineken', 'Intel', 'Kappa', 'Lining', 'McDonald', 'NB', 'Nike', 'Pepsi', 'Puma', 'Xtep']

实验截止目前，本文将YOLOv5系列五款不同参数量级的模型均进行了开发评测，接下来依次看下模型详情：

【yolov5n】

  1. # YOLOv5 
    声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Li_阴宅/article/detail/826161
    推荐阅读
    相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。