텍스트 파일2개로 나누기

import os
import shutil

path = 'D:/Yolo/darknet/build/darknet/x64/model/fuelhole04_combo'
train_txt = os.path.join(path, 'train.txt')
test_txt  = os.path.join(path, 'test.txt')

f = open(train_txt)
lines = f.readlines()
f.close()
#os.remove(train_txt)

f_train = open(train_txt, mode='wt', encoding='utf-8')
f_test  = open(test_txt, mode='wt', encoding='utf-8')

for i in range(len(lines)):
    if i % 10 == 0:
        f_test.write(lines[i])
    else:
        f_train.write(lines[i])

train.txt에 있는 파일 내용을

 

train.txt 90%, test.txt 10%의 비율로 나누기