[딥러닝] 사용자 데이터로 CNN 학습하기- 1. 라벨링하기

 

1. 라벨링하기

수집한 이미지 데이터에 라벨링 작업을 합니다.

 

_03_data_labelling.py

import os
import csv

dataDir='data.1695035538.287210' # 데이터 저장 디렉토리

print(os.getcwd()) # 현재 디렉터리 어딘지 확인(cwd=current working directory)
os.chdir(dataDir) # 디렉터리 이동(change directory)
roadDirs=os.listdir() # 현재 디렉터리 확인
print(roadDirs)

f_csv=open('0_road_labels.csv','w',newline='')
wr=csv.writer(f_csv)
wr.writerow(["file","label","labelNames"])

roadDirs=[road for road in roadDirs if os.path.isdir(road)]
print(roadDirs)

for num, roadDir in enumerate(roadDirs): # enumerate : 인덱스 번호까지 부여해줌
    roadFiles=os.listdir(roadDir)
    for roadFile in roadFiles:
        wr.writerow([os.path.join(roadDir, roadFile),num,roadDir]) # join : 합친다
        
f_csv.flush() 
f_csv.close()