|
|
|
|
|
''' |
|
@File : DataTools.py |
|
@Time : 2022/09/30 07:47:36 |
|
@Author : zzubqh |
|
@Version : 1.0 |
|
@Contact : baiqh@microport.com |
|
@License : (C)Copyright 2017-2018, Liugroup-NLPR-CASIA |
|
@Desc : 数据集预处理类 |
|
''' |
|
|
|
|
|
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() |
|
|
