机器学习
w747328720
这个作者很懒,什么都没留下…
展开
-
cuda+torch安装
1. 查看torch版本python -c "import torch; print(torch.__version__)"2. 查看cuda版本python -c "import torch; print(torch.version.cuda)"3、查看LInux系统配置nvidia-smiCUDA_VISBLE=0PYTHON4、安装cudatoolkit cudnn要写 CUDA 的程序,只需要安装 CUDA Toolkit 即可conda instal...原创 2021-12-08 16:36:43 · 3293 阅读 · 0 评论 -
nnUnet 马萨诸塞州道路分割 示例
https://github.com/MIC-DKFZ/nnUNet/blob/master/nnunet/dataset_conversion/Task120_Massachusetts_RoadSegm.py1、一定要先复制模型,因为每次训练不同数据集时,会在相同文件下进行操作,系统会报错。下载数据集road_segmentation_ideal2、找到程序Task120_Massachusetts_RoadSegm.py3\更改程序Task120_Massachusetts_..原创 2021-11-20 22:34:50 · 2523 阅读 · 1 评论 -
用SimpleITK查看医学图像
import SimpleITK as sitkfrom matplotlib import pyplot as plt def showNii(img): for i in range(img.shape[0]): plt.imshow(img[i,:,:],cmap='gray') plt.show() itk_img = sitk.ReadImage('F:\\BraTs learning\\brats2021\\BraTS2021_0.原创 2021-10-27 21:12:55 · 920 阅读 · 0 评论