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

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

• 项目效果：

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

(1) 安装

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) 训练

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

(3) 预测

predict3dunet --config <CONFIG>

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

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