Test / image.csv
Gabriele2027's picture
Create image.csv
ac51942 verified
raw
history blame contribute delete
895 Bytes
import csv
# Nome do arquivo CSV que será criado
csv_filename = 'dataset.csv'
# Lista de nomes de arquivo de imagem e seus respectivos rótulos
image_data = [
('IMG_6683.jpeg', 'label1'),
('IMG_6684.jpeg', 'label2'),
('IMG_6685.jpeg', 'label1'),
('IMG_6686.jpeg', 'label2'),
('IMG_6687.jpeg', 'label1'),
('IMG_6688.jpeg', 'label2'),
('IMG_6689.jpeg', 'label1'),
('IMG_6690.jpeg', 'label2'),
('IMG_6691.jpeg', 'label1'),
('IMG_6692.jpeg', 'label2'),
('IMG_6693.jpeg', 'label1'),
('IMG_6694.jpeg', 'label2'),
('IMG_6695.jpeg', 'label1')
]
# Criação do arquivo CSV
with open(csv_filename, mode='w', newline='') as file:
writer = csv.writer(file)
# Escreva o cabeçalho
writer.writerow(['filepath', 'label'])
# Escreva as linhas de dados
writer.writerows(image_data)
print(f"Arquivo {csv_filename} criado com sucesso.")