File size: 860 Bytes
a166479 |
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 |
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File : DataTools.py
@Time : 2022/09/30 07:47:36
@Author : zzubqh
@Version : 1.0
@Contact : [email protected]
@License : (C)Copyright 2017-2018, Liugroup-NLPR-CASIA
@Desc : 数据集预处理类
'''
# here put the import lib
import os
def create_annotations():
root_dir = r'/data/Dental'
img_dir = os.path.join(root_dir, 'img')
label_dir = os.path.join(root_dir, 'label')
annotation = 'tooth_label.md'
with open(annotation, 'w', encoding='utf-8') as wf:
for img_file in os.listdir(img_dir):
mask_file = os.path.join(label_dir, img_file.split('.')[0] + '_seg.nii.gz')
if os.path.exists(mask_file):
wf.write(f'{os.path.join(img_dir, img_file)},{mask_file}\r')
if __name__ == '__main__':
create_annotations()
|