python修改yolov4的调用load_meta

大家好我是新手 · 发表于 2022-7-4 07:34:08

本帖最后由大家好我是新手于 2022-7-4 07:42 编辑

load_meta = lib.get_metadata
lib.get_metadata.argtypes = [c_char_p]
lib.get_metadata.restype = METADATA
此函数加载分类列表,通过传递data文件返回分类结构体,data文件只是个跳板,真实读取的是names文件的内容,修改一下让直接读取names内容每次部署就可以少个文件并且就不需要配置data文件内的文件路径了。

def load_meta_py(name_path):
    with open(name_path,'r',encoding='utf-8') as f:
        class_names = f.read().splitlines()
    class_names = [i for i in class_names if i != '']
    charptr = POINTER(c_char_p)
    meta = METADATA(len(class_names),charptr(charptr(c_char_p(b''))))
    for x in range(len(class_names)):
        meta.names[x] = class_names[x].encode("ascii")
    meta.classes = len(class_names)
    return meta

rzbhelx · 发表于 2022-7-16 23:18:55

谢谢分将，不错不错

jnjcsh · 发表于 2022-7-7 14:51:13

666666666666666666666666

		自动登录	找回密码
密码			注册

[技术专题] python修改yolov4的调用load_meta