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

最新推荐文章于 2023-06-11 17:48:53 发布

原创最新推荐文章于 2023-06-11 17:48:53 发布 · 586 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#神经网络 #人工智能 #深度学习

python 专栏收录该内容

6 篇文章

订阅专栏

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

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)