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>
相关推荐
汉克老师17 分钟前
第十四届蓝桥杯青少组C++选拔赛[2023.2.12]第二部分编程题(4、最大空白区)
c++·算法·蓝桥杯·蓝桥杯c++·c++蓝桥杯
共享家952723 分钟前
优先搜索(DFS)实战
算法·leetcode·深度优先
一只懒洋洋42 分钟前
中值滤波、方框滤波、高斯滤波、均值滤波、膨胀、腐蚀、开运算、闭运算
算法·均值算法
shellvon1 小时前
你怎么被识别的?从TLS到Canvas的设备追踪术
后端·算法
薛定谔的算法1 小时前
JavaScript栈的实现与应用:从基础到实战
前端·javascript·算法
羚羊角uou1 小时前
【Linux】匿名管道和进程池
linux·c++·算法
空白到白2 小时前
决策树-面试题
算法·决策树·机器学习
flashlight_hi2 小时前
LeetCode 分类刷题:2563. 统计公平数对的数目
python·算法·leetcode
西猫雷婶2 小时前
scikit-learn/sklearn学习|广义线性回归损失函数的基本表达式
深度学习·神经网络·学习·机器学习·线性回归·scikit-learn·概率论
前端世界2 小时前
HarmonyOS 数据处理性能优化:算法 + 异步 + 分布式实战
算法·性能优化·harmonyos