41、利用分割技术查找疑似结节

利用分割技术查找疑似结节

1. 构建2D变换矩阵

在处理数据时， _build2dTransformMatrix 函数用于创建我们所需的变换矩阵。以下是该函数的代码：

def _build2dTransformMatrix(self):
    transform_t = torch.eye(3)
    for i in range(2):
        if self.flip:
            if random.random() > 0.5:
                transform_t[i,i] *= -1
    if self.rotate:
        angle_rad = random.random() * math.pi * 2
        s = math.sin(angle_rad)
        c = math.cos(angle_rad)
        rotation_t = torch.tensor([
            [c, -s, 0],
            [s, c, 0],
            [0, 0, 1]
        ])
        transform_t @= rotation_t
    return transform_t

这个函数会创建一个3×3的矩阵，后续可能会去掉最后一行。它会对2D数据进行增强操作，随机选择是否翻转，若进行旋转则会生成一个随机角度的旋转矩阵并应用到变换矩阵上。

GPU增强代码与CPU增强代码非常相似，这意味