PaddleOCR的rec如何裁剪数据

最新推荐文章于 2024-09-26 14:32:01 发布

一览天下945

最新推荐文章于 2024-09-26 14:32:01 发布

阅读量586

点赞数 3

分类专栏： PaddleOCR 文章标签： python opencv 计算机视觉

本文链接：https://blog.csdn.net/qq_38500228/article/details/125447453

版权

本文介绍了如何使用Python和OpenCV对PaddleOCR的训练数据进行裁剪。重点强调了train、test和val数据集需分三次运行代码，并注意相应的代码注释管理，以确保数据集划分的准确性。

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

PaddleOCR的rec如何裁剪数据为train,test,val并保存数据到label文件中

代码如下
- 要注意train,test,val是分三次运行代码的，每次要把对应的部分的代码注释去掉，剩下的部分用注释遮住，对应关系要正确

代码如下

要注意train,test,val是分三次运行代码的，每次要把对应的部分的代码注释去掉，剩下的部分用注释遮住，对应关系要正确


import json
import os
import numpy as np
import cv2


def get_rotate_crop_image(img, points):
    '''
    img_height, img_width = img.shape[0:2]
    left = int(np.min(points[:, 0]))
    right = int(np.max(points[:, 0]))
    top = int(np.min(points[:, 1]))
    bottom = int(np.max(points[:, 1]))
    img_crop = img[top:bottom, left:right, :].copy()
    points[:, 0] = points[:, 0] - left
    points[:, 1] = points[:, 1] - top
    '''
    assert len(points) == 4, "shape of points must be 4*2"
    # 求范数，得到宽度
    img_crop_width = int(
        max(
            np.linalg.norm(points[0] - points[1]),
            np.linalg.norm(points[2] - points[3])))
    # 求范数，得到高度
    img_crop_height = int(
        max(
            np.linalg.norm(points[0] - points[3]),
            np.linalg.norm(points[1] - points[

最低0.47元/天解锁文章