搜索结果
查询Tags标签： pytorch，共有 396条记录-
C++部署pytorch模型
https://zhuanlan.zhihu.com/p/191569603[toc] C++部署pytorch模型 前言 项目需要将pytorch训练好的网络用c++调用，在正式开始项目之前，在网上查了各种资料，共有三种实现方法： 直接将网络从最基础的CNN模块用C++实现； 将网咯模型和参数保存，然后使用opencv的DNN模…
2021/6/16 12:23:00 人评论 次浏览 -
Anaconda下安装pytorch（电脑同时存在两个python的情况）
首先在官网中运行Run this command后面的代码，全部安装成功后，在python交互环境下，输入import torch ,出现了如下错误：No module named numpy.core._multiarray_umathWhat??? 在网上找了找答案，说的是numpy版本太低，需要更新，于是我： pip install -i https://…
2021/6/14 12:24:36 人评论 次浏览 -
CBAM：通道注意力+空间注意力【附Pytorch实现】
论文地址：https://openaccess.thecvf.com/content_ECCV_2018/papers/Sanghyun_Woo_Convolutional_Block_Attention_ECCV_2018_paper.pdf1、动机 卷积操作是通过混合通道和空间两个维度的信息来间特征提取的。在注意力方面，SE仅关注了通道注意力，没考虑空间方面的注意力…
2021/6/12 10:22:59 人评论 次浏览 -
Windows 安装 pytorch3d
首先安装pytorch，我安装的是 pytorch 1.6.0 py3.7_cuda102_cudnn7_0 pytorch 1.下载pytorch3d到本地，主要不要下载master分支的，下载release的版本 ，我这里下载最新的 v0.2.0，下载链接是 https://github.com/facebookresearch/pytor…
2021/6/12 7:21:25 人评论 次浏览 -
语义分割学习笔记2——Jetson Xavier安装pytorch
语义分割学习笔记2——Jetson Xavier安装pytorch 一、whl文件安装pytorch二、安装torchvision三、测试是否安装成功运行语义分割的代码，需要先安装pytorch，下面是我在Xavier上安装语义分割的步骤。 我的环境说明： 名称版本硬件：Nvidia Jetson Xavier AGXCUDA版本：10.…
2021/6/7 10:26:49 人评论 次浏览 -
【图像分类】——来来来，干了这碗EfficientNet实战（Pytorch）
目录 摘要 新建项目 导入所需要的库 设置全局参数 图像预处理 读取数据 设置模型 设置训练和验证 测试 完整代码：摘要 EfficientNet是谷歌2019年提出的分类模型，自从提出以后这个模型，各大竞赛平台常常能看到他的身影，成了霸榜的神器。下图是EfficientNet—B0模型的网…
2021/6/5 10:26:21 人评论 次浏览 -
PyTorch 1.5上线：加入稳定C++前端，高级自动梯度API
点击上方 "程序员小乐"关注, 星标或置顶一起成长 每天凌晨00点00分, 第一时间与你相约 每日英文 Never expect. Never assume. And never demand. Just let it be. If its meant to be, it will happen. 不要期待，不要假想，不要强求，顺其自然，如果注定，便一…
2021/6/4 12:23:18 人评论 次浏览 -
Pytorch多卡训练
前一篇博客利用Pytorch手动实现了LeNet-5，因为在训练的时候，机器上的两张卡只用到了一张，所以就想怎么同时利用起两张显卡来训练我们的网络，当然LeNet这种层数比较低而且用到的数据集比较少的神经网络是没有必要两张卡来训练的，这里只是研究怎么调用两张卡。 现有方法…
2021/5/25 18:15:52 人评论 次浏览 -
Coordinate Attention +resnet+pytorch实现
# CA (coordinate attention)import torch import torch.nn as nn import math import torch.nn.functional as F from torchsummary import summary import torch.utils.model_zoo as model_zoo__all__ = [ResNet, resnet18, resnet34, resnet50, resnet101,resnet152]mo…
2021/5/25 10:27:32 人评论 次浏览 -
Pytorch学习
torch.nn.Conv2d 初始化模型：conv = torch.nn.Conv2d(1, 3, kernel_size=3, stride=2, padding=1)第一个参数：表示输入的通道数 第二个参数：表示输出的通道数 第三个参数：表示卷积核大小 第四个参数：表示步幅长度 第五个参数：是否需要padding调用模型：out = conv(x…
2021/5/23 10:27:12 人评论 次浏览 -
Windows环境下PyTorch_geometric安装踩坑
Windows环境下PyTorch_geometric安装踩坑 pytorch geometric在windows环境下安装非常恶心，莫名其妙各种报错。本帖针对GCC的编译error提供解决方案。一般报错的包主要是两个： torch-scattertorch-sparse Building wheels for collected packages: torch-scatterBuildin…
2021/5/23 7:29:24 人评论 次浏览 -
PyTorch程序设计
PyTorch程序设计 1. 程序示例2. 加载数据3. 创建网络（BN、卷积、ReLU）4. 优化器（优化器比较、参数lr、权重衰减）5. 损失函数比较6. 模型保存与加载1. 程序示例 import torch, os from torch.utils.data import Dataset, DataLoader import numpy as npclass MyDataSet…
2021/5/22 1:55:50 人评论 次浏览 -
Anaconda配置python3.6+pytorch(杂七杂八)
Anaconda配置python3.6+pytorch 1、安装Anaconda 由于开源软件镜像站清华貌似已经不提供Anaconda的镜像服务了，所以直接官网下吧，在梯子加持下速度还是很快的 2、配置python3.6环境 很多博客大都使用conda prompt或者cmd 调用conda create来创建环境，那Anaconda Naviga…
2021/5/16 22:25:24 人评论 次浏览 -
pytorch安装
2021/5/16 18:27:46 人评论 次浏览 -
pytorch学习之cycleGAN
cycleGAN实现马到斑马，照片到画像…的转换 原文地址：https://arxiv.org/pdf/1703.10593.pdf图1:给定任意两个无序图像集合X和Y，我们的算法学会自动“翻译”一张图像到另一张图像，反之亦然:(左)莫奈绘画和风景照片;(中)斑马和马图片;(右)来自Flickr的夏季和冬季约塞米蒂…
2021/5/11 18:28:54 人评论 次浏览
