python 批量提取txt里标记坐标 opencv截取对应区域并保存

最新推荐文章于 2024-08-10 08:22:26 发布

不当码农好不好

最新推荐文章于 2024-08-10 08:22:26 发布

阅读量822

点赞数

分类专栏：批量文件处理 python 文章标签： python opencv 脚本语言 txt

本文链接：https://blog.csdn.net/JZJZ73/article/details/107757952

版权

本文介绍了如何使用Python结合OpenCV，从TXT文件中提取坐标信息，生成图像的截取区域，并进行保存。具体步骤包括：1) 从TotalText数据集的TXT文档中筛选坐标；2) 计算坐标范围，形成包围框；3) 实现代码操作。

摘要由CSDN通过智能技术生成

1、全文根据TotalText数据集进行处理，txt文档打开后格式：

每一个line只有前两个元素x，y为所需坐标，故只对x，y进行处理

2、基本思路为提取x，y中最大最小的坐标形成一个box，每个box则包含所有的标记

3、code如下：

# -*- coding: UTF-8 -*-
#根据数据集批量截取图片并保存
import numpy as np
import glob as glob
import cv2
import os
from PIL import Image
import os
import time
import matplotlib.pyplot as plt

def main():
    path_txt = "～/下载/TotalText/examples/txt/traintrain/"
    count=1
    for f in os.listdir(path_txt):

        f = open(path_txt + f,'r')
        lines = f.readlines()
        num = len(lines)
        # i=0
        num = 1
        for line in lines:
            print(f)
            li = line.split(',')
            x = li[0].lstrip('x:').lstrip(' ').replace('[', '').replace('