Spaces:
Sleeping
Sleeping
File size: 972 Bytes
1cc0005 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
def win_ops():
data = ""
with open("D:\\Repos\\LungTumorSegmentation\\Resources\\train_data_win.yaml", "r") as f:
data = f.read()
data_split = data.split("label")
new_data = ""
ds = []
for i in data_split[3:]:
ds.append(i.split(",")[0].split("\\")[2].split("'")[0])
for n, i in enumerate(data_split[2:]):
new_data += i
if (n < 63):
new_data += f"boxes' : '{ds[n]}',\n 'label"
with open("D:\\Repos\\LungTumorSegmentation\\Resources\\train_data_win_with_boxes.yaml", "w+") as f:
f.write(new_data)
def ub_ops():
data = ""
with open("D:\\Repos\\LungTumorSegmentation\\Resources\\train_data_win_with_boxes.yaml", "r") as f:
data = f.read()
data = data.replace("\\" + "\\", "/")
with open("D:\\Repos\\LungTumorSegmentation\\Resources\\train_data_with_boxes.yaml", "w+") as f:
f.write(data)
if __name__ == "__main__":
ub_ops()
|