Pytorch
Along1617188
AI-follower
展开
-
Pytorch 常见问题总结
pytorch 常见问题总结原创 2022-08-10 09:20:46 · 327 阅读 · 0 评论 -
Pytorch 常用操作
Pytorch tensor操作原创 2022-08-02 20:24:10 · 131 阅读 · 0 评论 -
Python Hausdorff distance 豪斯多夫距离
github官方库https://github.com/deepmind/surface-distance代码示例import osimport numpy as npimport codecsimport SimpleITK as sitkimport pandas as pdimport torchimport surface_distance as surfdistdef readlines(file): """ read lines by removing .原创 2022-05-19 09:32:52 · 2191 阅读 · 0 评论 -
Pytorch dataset类
自己数据的dataset类,传入数据比如训练数据格式为.txt格式,字符串类型如下所示:/home/xx/data/volume_01.nii.gz /home/xx/data/label_01.nii.gz /home/xx/data/volume_02.nii.gz /home/xx/data/label_02.nii.gz /home/xx/data/volume_03.nii.gz /home/xx/data/label_03.nii.gz /home/xx/data/volume_04.原创 2022-01-27 15:16:32 · 2963 阅读 · 0 评论 -
Ubuntu20.04 cuda11.4 RuntimeError: CUDA error: no kernel image is available for execution on the dev
问题来源该问题源于项目工程自动下载的torch版本与cuda11.4支持的torch不兼容那么问题就在于如何找到cuda11.4兼容的torch版本?在官网中寻找一.选择与系统和cuda匹配的下载地址首先打开pytorch官网二.下载按照图示找到对应cuda的torch下载指令,需要注意的是这个链接https://download.pytorch.org/whl/cu113/torch_stable.html是一个下载地址的展示,而不是一个特定的下载地址,当把Run this Command输原创 2022-01-17 16:26:33 · 1299 阅读 · 0 评论 -
Python SimpleITK 计算医学图像dice
import osimport numpy as npimport codecsimport SimpleITK as sitkimport pandas as pdimport torch'''dice.txt context---len(num_images)---path of ground truth of image_001---path of seg_mask of image_001---path of ground truth of image_002---pa原创 2021-11-18 20:49:55 · 969 阅读 · 0 评论 -
Pytorch历史版本下载
Pytorch历史版本下载原创 2021-04-29 14:43:16 · 374 阅读 · 0 评论