Как преобразовать данные Contour структуры DICOM-RT в координату изображения? - PullRequest
0 голосов
/ 14 марта 2019

Я использую модуль Python-pydicom для загрузки образца пациента:

def load_data(sample):
    data=dict()
    # dcms data
    dcms_data=dict()
    for dcm_file in sample['dcm_files']:  # 遍历读取数据
        ds = pydicom.dcmread(dcm_file)
        array = ds.pixel_array
        origin = ds.ImagePositionPatient  # 网格原点在世界坐标系的位置
        spacing = ds.PixelSpacing  # 采样间隔
        uid = ds.SOPInstanceUID
        dcms_data[uid] = {'dcmSpacing': spacing, 'dcmOrigin': origin, 'array': array}
    data['dcms_data']=dcms_data

    # rt data
    rt_data = dict()  # 以{RUID:label_data}形式返回结果
    ds = pydicom.dcmread(sample['rt_file'])
    sequences = ds.ROIContourSequence[0].ContourSequence
    for sequence in sequences:
        ruid = sequence.ContourImageSequence[0].ReferencedSOPInstanceUID
        array = sequence.ContourData
        num = sequence.NumberOfContourPoints
        rt_data[ruid] = {'pointNumber': num, 'array': array}
    data['rt_data']=rt_data
    # 返回结果
    return data

Затем я преобразую данные структурного контура DICOM-RT в координату изображения:

def convert_global_aix_to_net_pos(data):
    point_data = {}  # 返回坐标{uid:data}
    for uid, value in data['rt_data'].items():
        num = value['pointNumber']
        label_data = value['array']
        dcm_origin = data['dcms_data'][uid]['dcmOrigin']
        dcm_spacing = data['dcms_data'][uid]['dcmSpacing']

        point = []  # 坐标[(x1,y1),(...),...]
        for i in range(0,num,3):
            x = label_data[i]  # 轮廓世界坐标系
            y = label_data[i + 1]
            X = int(float(x) - float(dcm_origin[0]) / float(dcm_spacing[0]))  # 轮廓X坐标
            Y = int(float(y) - float(dcm_origin[1]) / float(dcm_spacing[1]))  # 轮廓Y坐标
            point.append((X, Y))
        point_data[uid] = point
    return point_data

Однако, когда япротестировать эту функцию в моих файлах dicom. Я обнаружил, что она вернула неверные данные точек (отрицательные данные)

Я предполагаю, что метод преобразования данных контура DICOM-RT Struct в координаты изображения неверен, но я просто не могунайти другой путь.Мой метод неверен?Или как я мог это реализовать?Заранее спасибо.

1 Ответ

1 голос
/ 15 марта 2019

Похоже, вам не хватает скобок вокруг float(x) - float(dcm_origin[0]) (и аналогично для линии y).Вычитание должно быть выполнено до деления.

В противном случае все выглядит хорошо, предполагая ImageOrientationPatient из (1, 0, 0, 0, 1, 0).

...