- 博客(145)
- 资源 (6)
- 收藏
- 关注
转载 conda环境导出 | 安装库导出 到 environment.yaml 和 requirements.txt
conda环境导出 | 安装库导出
2022-06-25 21:41:03
1518
原创 解决pytorch多GPU训练的模型加载问题
由于多GPU训练使用了 nn.DataParallel(net, device_ids=gpu_ids) 对网络进行封装,因此在原始网络结构中添加了一层module。网络结构如下:而不使用多GPU训练的网络结构如下:重点多GPU训练好的模型,单GPU 如何正确加载方法一:(训练的时候必须写好的不同保存模式)或者写入字典方法二:无论有几个GPU 都按并行的方式加载即可,一个也可以,不影响或者字典中的加载参考链接:[1]https://blog.csdn.net/anshiquanshu/a
2022-06-07 20:14:06
685
原创 pytorch 实现预训练模型加载VGG16
1: 下载的预训练模型# VGG-16class UNet(nn.Module): def __init__(self, n_channels, n_classes, bilinear=True, pretrained=True): super(UNet, self).__init__() self.n_channels = n_channels self.n_classes = n_classes # 类别1 self.bili
2022-05-25 11:22:26
1536
转载 tensorboard 在 Linux服务器上的本地使用方法
1. 应有tensorboard生成文件形如: 2. 应有tensorboard: pip install tensorboard1 步骤 1. 打开一个本地session,以下为MobaXterm中显示: 2. 输入命令: ssh -L 16006:127.0.0.1:6006 <服务器登录用户名>@<服务器地址>1 例如: ssh -L 16006:127.0.0.1:6006 aaa@111.11.31.1111 其中: 1)替换端口名、服务器登录用
2022-05-18 16:18:55
1366
1
转载 卷积核类型简介
卷积核类型简介一个简短的介绍卷积使用“kernel”从输入图像中提取某些“特征”。kernel是一个矩阵,可在图像上滑动并与输入相乘,从而以某种我们期望的方式增强输出。看下面的GIF。上面的kernel可用于锐化图像。但是这个kernel有什么特别之处呢?考虑下图所示的两个输入图像。第一个图像,中心值为3 * 5 + 2 * -1 + 2 * -1 + 2 * -1 + 2 * -1 =7,值3增加到7。第二个图像,输出是1 * 5 + 2 * -1 + 2 * -1 + 2 * -1 + 2 * -1
2022-05-16 17:15:06
1283
转载 pytorch实现残差网络RestNet
pytorch_5.11 残差网络--RestNet </h2> <div class="postbody"> <div id="cnblogs_post_body" class="blogpost-body cnblogs-markdown">了解残差网络ResNet是何凯明在2015年提出的一种网络结构ResNet又名残差神经网络,指的是在传统卷积神经网络中加入残差学习(res...
2022-04-26 15:36:51
643
原创 OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.报错
添加程序(一定要添加在包的最前面)import osos.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
2022-04-25 15:25:35
849
原创 解决错误TypeError: float() argument must be a string or a number, not ‘_NoValueType‘_
pytorch 中常见的numpy版本问题报错,本人深受困扰以下三个问题都是numpy 包版本的问题(本人遇到:2,3):1:UserWarning: Failed to initialize NumPy: No module named ‘numpy.core._multiarray_umath‘2: RuntimeError: implement_array_function method already has a docstring3:TypeError: float() argument
2022-04-23 22:26:05
6122
原创 python实现去除图片纯色边框和给图片补边
# encoding: utf-8# author: qbit# date: 2020-06-15# summary: 去除图片纯色边框import shutilfrom PIL import Image, ImageChopsImage.MAX_IMAGE_PIXELS = 10000000000import numpy as npdef TrimImgEdge(inImgPath, outImgPath): r""" 去除图片边框 inImgPath: 输入图片路
2022-04-18 20:04:49
2966
MNIST手写数字识别 数据集
2020-11-04
Handbook of Semantic Web Technologies 2010.pdf
2019-09-27
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人