Python深度学习图片集转为CSV

import osimport csvdef create_csv(labels):path = '/Users/Desktop/pic/training/apple'name = os.listdir(path)with open(path + '.csv', 'w') as csvfile:writer = csv.writer(csvfile)writer.writerow(['data',

你怎么这么奇怪

1269人浏览 · 2021-06-26 12:10:21

你怎么这么奇怪 · 2021-06-26 12:10:21 发布

import os
import csv
def create_csv(labels):
    path = '/Users/Desktop/pic/training/apple'
    name = os.listdir(path)
    with open(path + '.csv', 'w') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(['data', 'label'])
        for n in name:
            if n[-4:] == '.jpg':
                print(n)
                writer.writerow(['/Users/Desktop/pic/training/' + str(labels) + '/' + str(n),'/Users/Desktop/pic/training/' + str(labels) + 'label/' + str( n[:-4] + '.png')])
            else:
                pass


if __name__ == "__main__":
    create_csv('testing')
    create_csv('training')