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       4 阅读
  2. rknn3588 yolov5 学习笔记

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

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

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

最近更新

  1. Android Display管理服务DMS

    2024-06-11 08:46:04       0 阅读
  2. CSS 媒体类型

    2024-06-11 08:46:04       0 阅读
  3. Spring框架的原理及应用详解(一)

    2024-06-11 08:46:04       0 阅读
  4. 数据库的隔离级别

    2024-06-11 08:46:04       0 阅读
  5. leetcode 70.爬楼梯

    2024-06-11 08:46:04       0 阅读
  6. 每周题解:Intervals(区间)

    2024-06-11 08:46:04       0 阅读
  7. dolphinscheduler独立集群部署文档(海豚调度)

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

热门阅读

  1. 递归

    递归

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

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

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

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