yolo中 LoadImages， LoadScreenshots， LoadStreams

dguochuan

于 2023-02-17 18:15:27 发布

阅读量2.6k

点赞数 1

文章标签： YOLO python 开发语言

本文链接：https://blog.youkuaiyun.com/qq_27172615/article/details/129091561

版权

该文介绍了一个名为LoadImages的类，用于读取本地图片、视频数据，支持自定义转换和步长。同时提到了LoadStreams，可以处理远程RTSP、RTMP和HTTP数据流，实现屏幕截图和远程流媒体的处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LoadImages：

功能：读取本地图片，视频，摄像头数据。

参数说明：

path: 路径地址（图片文件夹，文件，视频文件夹，文件，或者两者混合都可以。）

img_size: resize,或者letterbox转换后的大小。

stride: letterbox转换参数

auto: letterbox 转换参数

transforms: 自定义transforms

vid_stride: 视频文件fps步长

class LoadImages:
    def __init__(self, path, img_size=640, stride=32, auto=True, transforms=None, vid_stride=1):
        if isinstance(path, str) and Path(path).suffix == ".txt":  # *.txt file with img/vid/dir on each line
            path = Path(path).read_text().rsplit()
        files = []
        for p in sorted(path) if isinstance(path, (list, tuple)) else [path]:
            p = str(Path(p).resolve())
            if '*' in p:
                files.extend(sorted(glob.glob(p, recursive=True)))  # glob
            elif os.path.isdir(p):
                files.extend(sorted(glob.glob(os.path.join(p, '*.*'))))  # dir
            elif os.path.isfile(p):
                files.append(p)  # files
            else:

最低0.47元/天解锁文章