import numpy as np
import os
 
# modify train.txt/test.txt
dir_txt = "../data/ImageSets/{}.txt".format("test")
save_path = dir_txt

# labels from float to int
# print(list_txt)
# import pdb
# pdb.set_trace()
# import pdb
# pdb.set_trace()
txt_lis = []
with open(dir_txt, 'r') as fr:
    for line in  fr.readlines():
        # print(line)
        if line.split('_')[0] !='render':
            txt_lis.append(line)

with open(save_path, 'w') as fr:
    for line in txt_lis:
        fr.write("{}".format(line))