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>
相关推荐
何大春几秒前
【弱监督语义分割】Self-supervised Image-specific Prototype Exploration for WSSS 论文阅读
论文阅读·人工智能·python·深度学习·论文笔记·原型模式
福大大架构师每日一题2 分钟前
文心一言 VS 讯飞星火 VS chatgpt (396)-- 算法导论25.2 1题
算法·文心一言
uncle_ll8 分钟前
PyTorch图像预处理:计算均值和方差以实现标准化
图像处理·人工智能·pytorch·均值算法·标准化
EterNity_TiMe_17 分钟前
【论文复现】(CLIP)文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip
机器学习之心28 分钟前
一区北方苍鹰算法优化+创新改进Transformer!NGO-Transformer-LSTM多变量回归预测
算法·lstm·transformer·北方苍鹰算法优化·多变量回归预测·ngo-transformer
Suyuoa28 分钟前
附录2-pytorch yolov5目标检测
python·深度学习·yolo
yyt_cdeyyds39 分钟前
FIFO和LRU算法实现操作系统中主存管理
算法
alphaTao1 小时前
LeetCode 每日一题 2024/11/18-2024/11/24
算法·leetcode
kitesxian1 小时前
Leetcode448. 找到所有数组中消失的数字(HOT100)+Leetcode139. 单词拆分(HOT100)
数据结构·算法·leetcode
余生H2 小时前
transformer.js(三):底层架构及性能优化指南
javascript·深度学习·架构·transformer