Python-深度学习one-hot(独热)的txt标签自动制作

本文内容很简单,就是对已经获得的待分类图片名的txt文件进行ont-hot独热编码的添加,可以使用sublime text编辑器,alt+F3选中,然后进行独热编码,但是这种在数据量比较大的情况下,比较卡,这也是一个简单的方法,推荐。下面说另一种方法:

如题:原始图片名txt文件:

__author__ = "lingjun"
# E-mail: [email protected]
# 微信公众号:小白CV

with open("user.txt", "r") as f:
    for line in f.readlines():
        line = line.strip('\n')  #去掉列表中每一个元素的换行符
        txt = str(line)  + " 0 1 0 0" + "\n"
        print('%s ----> %s'%(line , txt))

        with open("user_new.txt", "a") as f:
            f.write(txt)

小白CV:公众号旨在专注CV(计算机视觉)、AI(人工智能)领域相关技术,文章内容主要围绕C++、Python编程技术,机器学习(ML)、深度学习(DL)、OpenCV等图像处理技术,深度发掘技术要点,记录学习工作中常用的操作,做你学习工作的问题小助手。只关注技术,做CV领域专业的知识分享平台。
————————————————
 

发布了74 篇原创文章 · 获赞 64 · 访问量 13万+

猜你喜欢

转载自blog.csdn.net/wsLJQian/article/details/91448488