30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
import os
|
|
# train_label_path = 'fire-8/train/labels'
|
|
# save_path = 'fire-8/train/newlabels'
|
|
# train_label_path = 'fire-8/test/labels'
|
|
# save_path = 'fire-8/test/newlabels'
|
|
|
|
train_label_path = 'fire-8/valid/labels'
|
|
save_path = 'fire-8/valid/newlabels'
|
|
all_train_txt_name = os.listdir(train_label_path)
|
|
all_train_txt = [os.path.join(train_label_path, t) for t in os.listdir(train_label_path)]
|
|
for i in range(len(all_train_txt)):
|
|
with open(all_train_txt[i], 'r') as f:
|
|
for line in f:
|
|
# print(line.split(' '))
|
|
line = line.split(' ')
|
|
if line[0] == '0':
|
|
line[0] = '2'
|
|
with open(os.path.join(save_path, all_train_txt_name[i]), 'a') as t:
|
|
t.write(' '.join(line))
|
|
continue
|
|
elif line[0] == '2':
|
|
line[0] = '3'
|
|
with open(os.path.join(save_path, all_train_txt_name[i]), 'a') as t:
|
|
t.write(' '.join(line))
|
|
continue
|
|
else:
|
|
print(all_train_txt_name[i])
|
|
continue
|
|
|
|
# break |