3D血管图像处理与连接系统

代码实现了一个基于血管图像图（Vessel Image Graph, VIG）的3D血管连接系统，主要用于处理和分析医学影像中的血管结构。以下是代码的主要功能：

import os
import torch
import torch.nn as nn
import torch.nn.functional as F
import numpy as np
import nibabel as nib
import matplotlib