YOLOv5的predict.py逐句讲解(学习笔记)

 

 因为太多依赖python的各种库,导致自己对YOLO的开发能力有所下降,最近准备重新整理一下YOLO系列的代码以供以后自己查阅。

YOLOv5-v7.0将分类脱离出来了。predict.py为分类的推理代码。predict.py主要有run(),parse_opt(),main()三个函数构成。

一、导入模块

这部分导入python模块与YOLO的自定义模块。

##############################################导入相关python模块##############################################
import argparse  # argparse模块的作用是用于解析命令行参数,例如python test.py --port=8080
import os  # os模块提供了非常丰富的方法用来处理文件和目录
import platform # platform模块用于获取操作系统的名称、版本号、位数等信息
import sys  # sys模块提供了对解释器使用或维护的一些变量的访问,以及与解释器强烈交互的函数,例如sys.exit()函数
from pathlib import Path # pathlib模块提供了一种对象化的路径操作方式,可以用来替代os.path模块,Path能更加方便对字符串路径进行操作
 
import torch  #pytorch框架
import torch.nn.functional as F  #pytorch框架中的函数库,主要用于

相关推荐

  1. YOLOv5predict.py逐句讲解学习笔记

    2024-06-11 08:46:04       23 阅读
  2. rknn3588 yolov5 学习笔记

    2024-06-11 08:46:04       21 阅读
  3. YOLOv7 学习笔记

    2024-06-11 08:46:04       45 阅读
  4. YOLOv6 学习笔记

    2024-06-11 08:46:04       50 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-11 08:46:04       5 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-11 08:46:04       5 阅读
  3. 在Django里面运行非项目文件

    2024-06-11 08:46:04       4 阅读
  4. Python语言-面向对象

    2024-06-11 08:46:04       6 阅读

热门阅读

  1. 递归

    递归

    2024-06-11 08:46:04      24 阅读
  2. OpenZeppelin Ownable合约 怎么使用

    2024-06-11 08:46:04       15 阅读
  3. 【Python】易错点——数组;列表;内存分配

    2024-06-11 08:46:04       14 阅读
  4. NLP--朴素贝叶斯

    2024-06-11 08:46:04       12 阅读