csv文件_训练图片标签_（个人学习记录）

最新推荐文章于 2022-12-11 14:59:24 发布

ZOY随笔

最新推荐文章于 2022-12-11 14:59:24 发布

阅读量528

点赞数 1

分类专栏：深度学习文章标签：深度学习

本文链接：https://blog.youkuaiyun.com/qq_35555072/article/details/115030071

版权

深度学习专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个Python脚本，用于从面部识别数据集中自动生成CSV文件，包含图像路径及其对应的标签。此外，还提供了一个命令行技巧来生成训练图片的路径列表。

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

csv文件_训练图片标签_记录

1、csv

opencv教程里面为我们提供了自动生成csv文件的脚本（create_csv.py）.
打开后是我们修改一下代码，BASE_PATH手动的改成自己的数据集路径，改完后代码如下：

#!/usr/bin/env python

import sys
import os.path

# This is a tiny script to help you creating a CSV file from a face
# database with a similar hierarchie:
#
#  philipp@mango:~/facerec/data/at$ tree
#  .
#  |-- README
#  |-- s1
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#  |-- s2
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#  ...
#  |-- s40
#  |   |-- 1.pgm
#  |   |-- ...
#  |   |-- 10.pgm
#

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print "usage: create_csv <base_path>"
        sys.exit(1)
   BASE_PATH="C:/Users/ZOY/Desktop/temp/video/orl_faces" 
    SEPARATOR=";"
    fh = open("C:/Users/ZOY/Desktop/temp/video/orl_faces/at.txt",'w')   //生成csv格式数据的路径
    label = 0

     for dirname, dirnames, filenames in os.walk(BASE_PATH):
        for subdirname in dirnames:
            subject_path = os.path.join(dirname, subdirname)
            for filename in os.listdir(subject_path):
                abs_path = "%s/%s" % (subject_path, filename)
                print("%s%s%d" % (abs_path, SEPARATOR, label))
                fh.write(abs_path)
                fh.write(SEPARATOR)
                fh.write(str(label))
                fh.write("\n")
            label = label + 1
    fh.close()