OCR/STR生僻字数据集生成 | PaddleOCR的垂类Fine-tune（1）

最新推荐文章于 2024-08-20 15:52:28 发布

訢詡

最新推荐文章于 2024-08-20 15:52:28 发布

阅读量1.4k

点赞数 1

分类专栏：深度学习NLP方向文章标签：机器学习算法 OCR STR 生僻字

本文链接：https://blog.csdn.net/Andrwin/article/details/120162193

版权

本文介绍如何利用特定字库生成包含生僻字的OCR训练图片，详细步骤包括导入依赖、配置函数、画图及生成函数，并遵循PaddleOCR自定义数据集格式记录图像文件信息。

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

根据字库生成一个图片，图片里面约10多个字，白底黑字，字号大约6号：

大概长这个样子

首先找个字库再说：https://github.com/elephantnose/characters

导入依赖包

import cv2
from PIL import ImageFont,ImageDraw,Image
import numpy as np
import uuid
import os
import random
import math

配置函数，画图函数，生成函数

def config():
    with open("word_dict2.txt") as f:
        word = f.readlines()
    word_dict_length = len(word)#字库字数
    fontPath = "PingFang Heavy.ttf"#黑体
    save_dir = "gen/images"
    min_words_length = 9
    max_words_length = 13
    img_count = 10
    return word,fontPath,save_dir,min_words_length,max_words_length,img_count


def draw_to_