![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
litchi&&mango
这个作者很懒,什么都没留下…
展开
-
nnUNet训练并推理自己的数据集
(默认所需环境已经配置好,且为linux环境与linux命令)(1) 安装hiddenlayer pip install --upgrade git+https://github.com/nanohanno/hiddenlayer.git@bugfix/get_trace_graph#egg=hiddenlayer(一行)(2)安装nnUNet1、在空间充足的位置安装nnUNet(最好超过100G),在该位置mkdir home2、然后进入home文件夹mkdir nnUNetFrame,进入原创 2022-02-07 16:24:25 · 9683 阅读 · 18 评论 -
GPU训练完后自动关机
GPU训练完后自动关机:在代码最末尾添加import osos.system("shutdown")原创 2022-01-28 09:57:23 · 1682 阅读 · 0 评论 -
终端下创建pytorch环境
conda create -n torch python=3.8conda activate(deactivate) torch装包:pip install sklearnpip install SimpleITKpip install numpypip install matplotlibpip install opencv-pythonpip install pandas# pytorch官网下载合适的版本,注意conda大版本要一致,如这里的11conda install p.原创 2022-01-07 23:24:35 · 1075 阅读 · 0 评论 -
多张png合成为nii.gz
将多张png图片按照(h, w, c)组装成nii.gzimport numpy as npimport nibabel as nibfrom glob import globfrom torch.utils.data import Dataset, DataLoaderfrom torchvision import transformsfrom PIL import Imageclass DataSet(Dataset): # 方便读数据 def __init__(self,原创 2021-08-26 17:05:32 · 1661 阅读 · 4 评论 -
pytorch生成包的版本文件及colab下打开终端
1.pipreqs复现别人的代码有时会由于包版本而造成结果有很大的误差,因此有时需要查看我们项目中使用的包的版本,在python中可以通过pipreqs直接查看;安装模块:pip install pipreqs之后直接在终端输入pipreqs即可,注意是要在自己运行后的项目中执行该命令,因为它是根据项目运行情况来提取出项目使用过的包,以及它们对应的版本;2.colab终端colab在pro模式下可以直接打开终端,但是普通模式就需要依赖kora包!pip install korafrom k原创 2021-08-19 21:21:33 · 711 阅读 · 0 评论 -
pytorch下使用netron进行网络结构可视化+下载colab的数据文件+下载kaggle的数据文件
https://github.com/lutzroeder/netron#installhttps://netron.app/原创 2021-08-19 20:35:40 · 656 阅读 · 0 评论