Pytorch
文章平均质量分 75
qq_37172182
i++, ++i, c++, g++, x++
展开
-
Android----Depth Anything尝鲜 小米手机部署
Depth Anything & Andorid部署原创 2024-08-04 23:44:48 · 463 阅读 · 0 评论 -
OnnxRuntime----Lite-Mono单目深度估计ONNX推理
Lite-Mono depth estimation OnnxRuntime原创 2023-04-16 23:29:43 · 1301 阅读 · 8 评论 -
VSCode配置之OnnxRuntime(CPU) && YOLOv7验证
VSCode与VS2022配置OnnxRuntime环境原创 2023-03-20 21:29:55 · 3213 阅读 · 12 评论 -
Ubuntu20.04 + RTX 3090(兼容RTX 2080 Ti) + Pytorch1.7配置方法
背景介绍:由于在Ubuntu16.04系统上安装RTX 3090显卡驱动有点吃力(各种Error和不兼容),使用最新Ubuntu20.04系统搭配最新的RTX 3090显卡配置最新的Pytorch【(*^▽^*)】前期准备: 1、Ubuntu20.04下载:Ubuntu20.4_amd64_desktop.iso 2、UNtebootin光盘刻录软件下载:unetbootin,选择Windows下载 3、NVID...原创 2020-11-30 16:50:56 · 2289 阅读 · 0 评论 -
Pytorch CAM特征可视化
背景: 类别激活映射(Class Activation Mapping, CAM)用于对深度学习特征可视化,通过特征响应定位图像的关键部位,为深度学习可解释性提供了一种方法,ACM以热力图的方式展示了图像局部响应的强弱信息,对应于更强的位置具有更好的特征识别能力。论文链接:Learning Deep Features for Discriminative LocalizationCAM基本原理: 定义类别分数,其中表示最后一个卷积层第通道的输出,为第个通道对应的类别的权重...原创 2020-10-09 19:54:00 · 3798 阅读 · 13 评论 -
图像读取库合集——cv2, PIL, skimage与numpy, pytorch(ToPILimage)
图像读取库合集——cv2, PIL, skimage与numpy, pytorch(ToPILimage)1 图像读取与属性1.1 PIL与numpy间的相互访问import numpy as npfrom PIL import Image#read a image with 3 channels, 500x889 pixelsimg_pil = Image.open('./test.png')#show a imageimg_pil.show()#get image imfopr原创 2020-10-05 11:19:02 · 1925 阅读 · 0 评论 -
Pytorch学习基础——损失函数
Pytorch学习基础——损失函数损失函数的形式化表示:Lloss=∑i=1Nbcriterion(y∗,y)L_{loss} = \sum_{i=1}^{N_b}criterion(y^{*},y)Lloss=∑i=1Nbcriterion(y∗,y), 其中y∗∈RB×Cy^{*}\in \R^{B\times C}y∗∈RB×C为模型预测输出,y∈RB×Cy\in \R^{B\times C}y∈RB×C为真实标签或标注,BBB为batch_sizebatch\_sizebatch_size,原创 2020-10-01 15:06:59 · 329 阅读 · 0 评论 -
Pytorch学习基础——RNN从训练到测试
在上一篇Pytorch学习基础——LeNet从训练到测试讲述了简单神经网络LeNet识别MNIST数据集的实例,作为对比,本次将结合LSTM实现对MNIST数据集的识别。实现过程:导入必要的包并设置超参数:import torchimport torchvisionfrom torch import nnfrom torch.autograd import Variableim...原创 2019-07-13 21:28:41 · 3899 阅读 · 1 评论 -
win10安装pytorch-geometric (PYG)包----亲测可用
pytorch安装参考:win10安装CUDA9.0+CUDANN+Anaconda3+PYTORCH1.xwin10下安装错误:$ pip install --verbose --no-cache-dir torch-scatter$ pip install --verbose --no-cache-dir torch-sparse$ pip install --verbose --no-cache-dir torch-cluster$ pip install --verbose --no原创 2020-05-19 10:17:54 · 2210 阅读 · 4 评论 -
TensorboardX----linux服务器远程训练,Windows本地查看日志
使用pytorch训练大型数据集是常常需要通过loss的下降曲线或者acc准确率的上升情况直观上判断模型的设计是否合理,使用tensorboardX将迭代的loss和acc加入scale中,方便查看中间过程,及时调整模型。安装:pip install tensorflowpip install tensorboardX定义:import numpy as npfr...原创 2020-02-18 12:08:50 · 2274 阅读 · 0 评论 -
Pytorch实例----DCGAN图像生成
背景: 深度学习对数据量的需求明显高于传统机器学习方法,当真实的数据量较少或难以满足实际网络收敛情况下,需用通过数据增强方法生成更多可用于训练的更多图片,数据增强方法通过对图像进行随机剪裁,翻转等变换,提升图像输入的丰富度,但数据增强本质上是在原图上做“线性变化”,无法产生原图中没有的图像。在这种条件下,GAN(Generative Adversarial Networks)能够...原创 2019-12-22 10:09:35 · 3533 阅读 · 8 评论 -
MANFOLD TX2配置Pytorch1.x方法总结(Python3.5+CUDA9.0+Opencv3.4.1)
硬件环境:MANFOLD TX2Ubuntu16.04Python 3.5CUDA 9.0CDUANN 7.x配置过程:1、查看CUDA与CUDANN版本dji@manifold2:~$ nvcc -Vnvcc: NVIDIA (R) Cuda compiler driverCopyright (c) 2005-2017 NVIDIA Corporation...原创 2019-11-18 22:24:42 · 745 阅读 · 3 评论 -
Pytorch实例----CAFAR10数据集分类(ShuffleNet)
在上一篇Pytorch实例----CAFAR10数据集分类(ResNet)的识别统计,本篇主要调整Net()类,设计ShuffleNet轻量级网络(+BN),实现对CAFAR10数据集的分类任务。ShuffleNet网络结构编程实现:#define shuffle blockclass ShuffleBlock(nn.Module): def __init__(self, g...原创 2019-11-12 18:09:42 · 811 阅读 · 0 评论 -
Pytorch实例----CAFAR10数据集分类(ResNet)
在上一篇Pytorch实例----CAFAR10数据集分类(VGG)的识别统计,本篇主要调整Net()类,设计ResNet网络(+BN),实现对CAFAR10分类数据集的分类任务。ResNet网络结构编程实现:#create residual blockclass ResidualBlock(nn.Module): def __init__(self, inchannel, ...原创 2019-11-10 10:45:05 · 1541 阅读 · 0 评论 -
Pytorch学习基础——torch.DataLoader
Pytorch官方文档:API for pytorchDataLoader函数功能:生成数据集的可迭代对象; 利用多线程加速batch data处理; 简洁、高效、直观的用于网络输入的数据结构,使用灵活,便于扩展DataLoader类位于torch.utils.data包下,官方API介绍如下:常用参数说明:dataset(Dataset):输入数据集 batch_si...翻译 2019-07-05 21:18:32 · 722 阅读 · 0 评论 -
Pytorch学习基础——torch.squeeze() & torch.unsqueeze()
Pytorch官方API:torch.squeeze()squeeze()函数功能:主要对数据的维度进行压缩(默认为1)。也可以通过dim指定位置,删掉指定位置的维数。unsqueeze()函数功能:对数据维度进行扩充。dim指定位置,添加指定位置的维数添加1。常见参数介绍:input(Tensor):输入张量; dim(int, optional):设定(可选)维度...翻译 2019-07-05 21:38:58 · 885 阅读 · 0 评论 -
Pytorch实例----CAFAR10数据集分类(AlexNet)
CAFAR10数据集介绍:CAFAR图像大小:训练集:50000测试集:10000类别:classes = ('plane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')实验步骤:1、加载数据集:torchvision.dataset.CIFAR10(p...原创 2019-08-19 08:20:19 · 1860 阅读 · 0 评论 -
win10安装CUDA9.0+CUDANN+Anaconda3+PYTORCH1.x
当前神经网络方法的训练和预测越来越依赖于GPU并行运算,因此在win10上配置CUDA+Pytorch,记录配置过程。硬件平台: win10(64位), NVIDIA Geforce 940M软件下载: 1:CUDA9.0,安装时注意加载路径 (环境变量)官网地址:CUDA9.0下载 2:CUDANN,对应CUDA9.0版本的CUDANN官网地址:...原创 2019-08-14 09:13:29 · 1406 阅读 · 0 评论 -
Pytorch实例----CAFAR10数据集分类(VGG)
在上一篇 Pytorch实例----CAFAR10数据集分类(AlexNet)的识别统计,本篇主要调整Net()类,设计VGG网络(+BN)后的识别统计(其他设计注释同上)。 VGG与AlexNet在CAFAR10数据集的统计结果对比图可以看到,对于之前cat(19%...原创 2019-08-23 19:36:57 · 959 阅读 · 0 评论 -
Pytorch学习基础——CNN基本结构搭建
Pytorch CNN结构特点:面向对象编程,即网络模型类继承自nn.Module基类; 重写构造函数和forward函数; 定义分类器classifier,最后做全连接处理;Pytorch CNN实例:*根据LeNet5的结构模型编写LeNet网络:图像经过卷积、池化等步骤的尺寸计算方式如下:图像的尺寸为,其中为width, h为height,c为channel;卷...翻译 2019-07-13 11:38:57 · 3447 阅读 · 0 评论