- 项目应用场景
- 用于 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
