nnDetection安装到使用,包括环境搭建,数据预处理和训练(没有使用docker)

nnDetection安装和使用
服务器环境:CUDA11.4 +torch1.12+24G显卡+linux
网上有挺多nnunet的安装和使用教程,为啥没有nndetection的教程呢,是因为太新了吗
用了这麽久才知道setup.py是这样用,是真的菜
在这里插入图片描述
安装conda,新建自己的环境,安装torch及其依赖,之后下载nndetection完整项目文件并解压,之后cd到包含setup.py的目录,输入pip install -e.安装nndetection所需依赖包。
在这里插入图片描述
看到进度条和successful心情就好,哈啊哈哈
环境装完了,就开始代码呗!
之前弄过nnunet,nndet也一样,先数据处理。运行preprocess.py,先报错环境变量没有设置好,包括det_data和det_models两个变量,那就先设置环境变量吧,参考博客https://zhuanlan.zhihu.com/p/359354934设置。我的环境是建在硬盘上的,因为内存有限,而且个人数据集比较大,所以用于保存数据的路径det_data和det_models都是指向硬盘的。
之后是数据集格式的问题,各种问题,无奈只好先试一下推荐的案例了
直接终端运行nnunet_example生成数据集
在这里插入图片描述
生成的数据集路径显示为,这个文件夹和路径后面要参考来创建自己的文件夹并应用到自己的数据集上。
在这里插入图片描述
其中生成的文件包括raw_splitted和dataset.json和logging.log
之后运行在这里插入图片描述运行nndet_prep 000D3_Example进行数据处理。
在这里插入图片描述
接下来换到我自己的数据集上:
在这里插入图片描述
这是我的数据集,命名方式如下
在这里插入图片描述
上面是labelsTs
在这里插入图片描述
上面是labelsTr
在这里插入图片描述
上面是imageTs
在这里插入图片描述
上面是imagesTr
在这里插入图片描述
这个是raw_splitted文件夹里的dataset.json文件内容
在这里插入图片描述
这个是标签labelsTr和labelsTs里面每个case对应的json文件内容。这个可以查看nndetection官方代码readme中的json文件内容制作属于自己的json文件。
之后进入数据预处理过程
在这里插入图片描述
所需时间较长,请耐心等待,我的数据较多,时间更长
预处理之后生成的文件和example一样,这是官网给的文件明细
在这里插入图片描述之后是nndet_unpack
在这里插入图片描述这个unpack是将npz文件转换为npy,之前没有做过目标检测相关的任务,具体为啥要转换还不清楚。
之后就是训练过程了
在这里插入图片描述
在这里插入图片描述
训练过程比较慢,一个epoch跑两三个小时也是正常的,请耐心等待。
我在跑训练的时候遇到过
在这里插入图片描述
跑到这里就卡死了,纹丝不动,折磨了好久,最后在官网上找到了作者的解答,在这里插入图片描述
原因是环境变量OMP_NUM_THREADS=1,要设置,不然会出现奇怪的问题。

  • 4
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 24
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 24
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值