일지

[Python] 텍스트 파일 한줄씩 리스트로 읽기

Jinsoo Kim 2020. 9. 16. 14:37
반응형

yolov4.names
0.00MB

 

 

텍스트 파일로 저장된 클래스명을 변수로 읽는 방법

 

< 소스 코드 >

f = open('D:/blog/coco.names', 'r')
data = f.read()

 

 

< 결과 >

클래스가 나뉘지 않고 하나의 문자열로 읽히게 됩니다.

 

텍스트 파일을 한줄씩 리스트로 저장하는 방법

 

< 소스 코드 >

with open('yolov4.names') as f:
    data = [line.strip() for line in f.readlines()]

 

< 결과 >

 

YOLO의 클래스를 80개의 리스트로 읽는 것을 확인할 수 있습니다.