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

  • 项目应用场景
    • 用于 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>
相关推荐
Dongsheng_20191 小时前
【汽车篇】AI深度学习在汽车零部件外观检测——机电轴承的应用
人工智能·深度学习·汽车
努力学习的小廉1 小时前
我爱学算法之—— 模拟(下)
c++·算法
en-route2 小时前
从零开始学神经网络——LSTM(长短期记忆网络)
人工智能·深度学习·lstm
海琴烟Sunshine3 小时前
Leetcode 26. 删除有序数组中的重复项
java·算法·leetcode
PAK向日葵3 小时前
【算法导论】NMWQ 0913笔试题
算法·面试
PAK向日葵3 小时前
【算法导论】DJ 0830笔试题题解
算法·面试
PAK向日葵3 小时前
【算法导论】LXHY 0830 笔试题题解
算法·面试
努力毕业的小土博^_^3 小时前
【深度学习|学习笔记】详细讲解一下 深度学习训练过程中 为什么 Momentum 可以加速训练?
人工智能·笔记·深度学习·学习·momentum
清风吹过3 小时前
少样本学习论文分享:多模态和类增量学习
论文阅读·人工智能·深度学习·学习·机器学习
麦麦麦造4 小时前
DeepSeek突然发布 V3.2-exp,长文本能力加强,价格进一步下探
算法