pytorch-3dunet
pytorch-3dunet实现了多种3D U-Net模型及其变体,包括标准3D U-Net、残差3D U-Net和带压缩激励块的残差3D U-Net。该项目支持二元和多分类语义分割以及去噪、学习反卷积等回归问题。项目还支持2D U-Net,提供多种配置示例帮助用户训练和预测。此外,该项目可在Windows和OS X系统上运行,并支持多种损失函数和评估指标,如Dice系数、平均交并比、均方误差等。这一描述更加简洁、流畅,同时保持了准确性。