IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> yolov5 test.py val.py detec.py 区别在哪里呢? -> 正文阅读

[人工智能]yolov5 test.py val.py detec.py 区别在哪里呢?

yolov5 test.py val.py detec.py 区别在哪里呢?

用户在训练数据的时候必须使用 train.py 来进行 数据训练和验证,但我很难理解detect.py 和test.py 之间的区别。应该在一个数据集中的(看不见的)测试数据上运行这两者中的哪一个?

首先: test.py 最近被重命名为 val.py!

这 3 个文件是为不同目的而设计的,并使用具有不同设置的不同数据加载器。

  • train.py 数据加载器旨在兼顾速度和准确性,
  • val.py 旨在获得验证数据集上的最佳 mAP,
  • detect.py 旨在真实世界中获得最佳的推理结果。

每个文件中的几个重要方面包括:

train.py

trainloader: LoadImagesAndLabels():旨在加载训练数据集图像和标签。增强能力可以选择启用。

 # Trainloader 
 dataloader, dataset = create_dataloader(train_path, imgsz, batch_size, gs, opt, 
                                         hyp=hyp, augment=True, cache=opt.cache_images, rect=opt.rect, rank=rank, 
                                         world_size=opt.world_size, workers=opt.workers, 
                                         image_weights=opt.image_weights, quad=opt.quad, prefix=colorstr('train: ')) 

testloader: LoadImagesAndLabels():旨在加载 val 数据集图像和标签。增强能力但被禁用。

 testloader = create_dataloader(test_path, imgsz_test, batch_size * 2, gs, opt,  # testloader 
                                hyp=hyp, cache=opt.cache_images and not opt.notest, rect=True, rank=-1, 
                                world_size=opt.world_size, workers=opt.workers, 
                                pad=0.5, prefix=colorstr('val: '))[0] 
  • image size: 640
  • rectangular inference: False
  • confidence threshold: 0.001
  • iou threshold: 0.6
  • multi-label: True
  • padding: None

val.py

dataloader: LoadImagesAndLabels():设计用于加载训练、验证、测试数据集图像和标签。增强能力但被禁用。

 dataloader = create_dataloader(data[task], imgsz, batch_size, gs, single_cls, pad=0.5, rect=True, 
                                prefix=colorstr(f'{task}: '))[0] 
  • image size: 640
  • rectangular inference: True
  • confidence threshold: 0.001
  • iou threshold: 0.6
  • multi-label: True
  • padding: 0.5 * maximum stride

detect.py

dataloaders(多个):设计用于加载多种类型的媒体(images, videos, globs, directories, streams)。

 # Set Dataloader 
 vid_path, vid_writer = None, None 
 if webcam: 
     view_img = check_imshow() 
     cudnn.benchmark = True  # set True to speed up constant image size inference 
     dataset = LoadStreams(source, img_size=imgsz, stride=stride) 
 else: 
     dataset = LoadImages(source, img_size=imgsz, stride=stride) 
  • image size: 640
  • rectangular inference: True
  • confidence threshold: 0.25
  • iou threshold: 0.45
  • multi-label: False
  • padding: None

YOLOv5 PyTorch Hub Inference

models.autoShape() 类用于图像加载、预处理、推理和 NMS。有关更多信息,请参阅 YOLOv5 PyTorch Hub 教程

 class autoShape(nn.Module): 
     # input-robust model wrapper for passing cv2/np/PIL/torch inputs. Includes preprocessing, inference and NMS 
     conf = 0.25  # NMS confidence threshold 
     iou = 0.45  # NMS IoU threshold 
     classes = None  # (optional list) filter by class 
  
     def __init__(self, model): 
         super(autoShape, self).__init__() 
         self.model = model.eval() 
  
     def autoshape(self): 
  • image size: 640
  • rectangular inference: True
  • confidence threshold: 0.25
  • iou threshold: 0.45
  • multi-label: False
  • padding: None
  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2021-12-23 15:46:04  更:2021-12-23 15:48:09 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年5日历 -2024/5/19 3:40:33-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码