YOLOv3生成中文标签

前提：本文的YOLOv3为AlexeyAB的版本。
step1：生成标签
生成标签的代码借鉴了https://www.pianshen.com/article/8064424145/
先在data/labels下创建python程序，写入以下代码，其中label.names为自定义分类的汇总，如
京
沪
晋
赣
…
G
H
Q
R
…
3
9
保存为txt，之后运行以下python程序

import os
import string
import pipes

font = 'futura-normal'
fileName=open("label.names",'r')
l=[label.lstrip().rstrip() for label in fileName.readlines()]


def make_labels(s):
    i = 0
    for word in l:
        os.system(
            "convert -fill black -background white -bordercolor white -border 4  -font /usr/share/fonts/truetype/arphic/ukai.ttc -pointsize %d label:\"%s\" \"%d_%d.png\"" % (
            s, word, i, s / 12 - 1))
        i = i + 1


for i in [12, 24, 36, 48, 60, 72, 84, 96]:
    make_labels(i)

step2：修改代码
生成了标签，之后修改代码
参见 https://blog.csdn.net/phinoo/article/details/89501569此处的修改
注意：他的标签生成程序可能有问题，我没有成功，所以我结合了两个人的博客实现了输出中文标签

step3：关键一步
最后关键一步，至少在我这里为关键一步

修改 **load_alphabet函数，修改为自己的识别类别，如何知道自己的识别类别呢，查看生成的标签，x_y.png，x为类别，y为字体大小

image **load_alphabet()
{
    int i, j;
    const int nsize = 8;
    image** alphabets = (image**)xcalloc(nsize, sizeof(image*));
    for(j = 0; j < nsize; ++j){
        alphabets[j] = (image*)xcalloc(128, sizeof(image));
        //将此处改为自己的标签类别，如67个类别，所以改为0-68
        for(i = 0; i < 67; ++i){
            char buff[256];
            sprintf(buff, "data/labels/%d_%d.png", i, j);
            alphabets[j][i] = load_image_color(buff, 0, 0);
        }
    }
    return alphabets;
}

结果如下，标签的大小修改还没有找到在哪里修改，如果有小伙伴实现了修改大小和线的粗细，可以留言告知一下，谢谢。