YOLOv5-4.0-google_utils.py 源代码导读

YOLOv5介绍

YOLOv5为兼顾速度与性能的目标检测算法。笔者将在近期更新一系列YOLOv5的代码导读博客。YOLOv5为2021.1.5日发布的4.0版本。
YOLOv5开源项目github网址
本博客导读的代码为utils文件夹下的google_utils.py文件，更新日期为2021.1.14.

google_utils.py

该文件负责从github/googleleaps/google drive等网站下下载所需要的的一些文件。
相关导入模块及说明如下所示。

#谷歌云对应的链接
# Google utils: https://cloud.google.com/storage/docs/reference/librariesimport os          #与操作系统交互的模块
import platform    #提供获取操作系统相关信息的模块
import subprocess  #子进程定义及操作的模块
import time        #获取系统时间的模块
from pathlib import Path  #使字符串路径易于操作的模块import requests   #通过urllib3实现自动发送HTTP/1.1请求的第三方模块
import torch      #pytorch模块

gsutil_getsize 函数用于返回网站链接对应文件的大小

def gsutil_getsize(url=''):# gs://bucket/file size https://cloud.google.com/storage/docs/gsutil/commands/du#创建一个子进程在命令行执行 gsutil du url命令s = subprocess.check_output(f'gsutil du {url}', shell=True).decode('utf-8')return eval(s.split(' ')[0]) if len(s) else 0  # 返回文件的byte大小

attempt_download 函数实现 从几个云平台下载预训练模型

def attempt_download(file, repo='ultralytics/yolov5'):#如果对应的文件不存在 则尝试下载# .strip()删除字符串前后的空格 /n /t等 .replace()将'替换为空格 .lower()将字符串全部转换为小写 Path()将str转换为Path对象file = Path(str(file).strip()

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！