进行分类任务的模型进行推理的时候,几个分类还好说,复杂的模型会有大量分类,如果这些分类在txt文件中:
手动复制粘贴会累死,几行代码可以搞定:
#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time : 2022/7/12 14:56
# @Author : coco
# @File : generate_classes.py
path = 'classes_name.txt'
with open(path, 'r', encoding='utf-8') as f:
texts = f.readlines()
classes = []
for text in texts[2:]:
data = text.strip().split(' ')
classes.append(data[0])
print(data)
print(classes)
print(len(classes))
结果展示:
选中属性的行,复制粘贴到推理代码中的class中 轻松解决。
有时候觉得努力没有用,但是如果把人生放在更长的角度去看,会有用的,你得有耐心,黎明前天是最黑的,只要再熬一点点时间,天就亮了 。——董宇辉
暑期编程PK赛 得CSDN机械键盘等精美礼品!


