读取txt文件内的预测框（神经网络）

王先生笔记

于 2021-12-24 09:47:10 发布

阅读量570

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签：神经网络人工智能深度学习

本文链接：https://blog.youkuaiyun.com/handsome_wang5/article/details/122113191

python 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个Python函数，用于从指定路径的文本文件中读取物体检测的标注框信息。该函数解析每一行的数据，提取类别名称及左上角和右下角的坐标，并将这些信息组织成列表返回。

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

import os


def read_txt(path):
    all_box = []
    with open(path, 'r') as f:
        while True:
            line = f.readline()
            if line:
                line = line.split("('")[-1]
                line = line.split(")")[0]
                line = line.replace("'", "")
                lists = line.split(", ")
                for i in range(4):
                    lists[i + 1] = int(lists[i + 1])
                all_box.append(lists)
            else:
                break
    return all_box


path = r"F:\pytorch\Annotations_pbox\000001.txt"
all_box = read_txt(path)
print(all_box)