3D分割项目 | 基于Pytorch+3DUnet实现的3D体积语义分割算法

极智视界2024-04-02 17:15

项目应用场景
- 用于 3D 体积语义分割场景，适用于各种物体的 3D 语义分割，比如大米、大豆的体积分割等

项目效果：

项目流程 ==> 具体参见项目内README.md

(1) 安装

bash 复制代码

conda install -c conda-forge mamba
mamba create -n pytorch-3dunet -c pytorch -c nvidia -c conda-forge pytorch pytorch-cuda=12.1 pytorch-3dunet
conda activate pytorch-3dunet

python setup.py install

(2) 训练

bash 复制代码

python evaluation/evaluate.py --config config/H36M.yaml

(3) 预测

bash 复制代码

predict3dunet --config <CONFIG>

(4) 也支持并行训练 & 并行预测

bash 复制代码

CUDA_VISIBLE_DEVICES=0,1 train3dunet --config <CONFIG>

# or

CUDA_VISIBLE_DEVICES=0,1 predict3dunet --config <CONFIG>

项目获取
- https://download.csdn.net/download/weixin_42405819/89039866