人工智能
-
迁移学习：keras + vgg16 + cifar10 实现图像识别
框架：keras数据集：CIFAR10模型：vgg16注：vgg16模型的输入图像尺寸至少为 48*48 思路：去掉vgg16的顶层，保留其余的网络结构与训练好的权重。然后添加模型结构，进而训练CIFAR10。 1.模型结构 2.具体代码以及注释 ①训练代码 #-*- coding: utf-8 -*- #迁移学习，vgg16+cifar10 from ke…
-
『PyTorch』屌丝的PyTorch玩法
1. prefetch_generator 使用 prefetch_generator库 在后台加载下一batch的数据，原本PyTorch默认的DataLoader会创建一些worker线程来预读取新的数据，但是除非这些线程的数据全部都被清空，这些线程才会读下一批数据。使用prefetch_generator，我们可以保证线程不会等待，每个线程都总有至少一…
-
PyTorch安装及试用 基于Anaconda3
设置Torch国内镜像 conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 安装PyTorch和TorchVision conda install pytorch torchvision 测试pytorch版本 impor…
-
基于pytorch框架的图像分类实践(CIFAR-10数据集)
在学习pytorch的过程中我找到了关于图像分类的很浅显的一个教程上一次做的是pytorch的手写数字图片识别是灰度图片，这次是彩色图片的分类，觉得对于像我这样的刚刚开始入门pytorch的小白来说很有意义，今天写篇关于这个图像分类的博客. 收获的知识 1.torchvison 在深度学习中数据加载及预处理是非常复杂繁琐的，但PyTorch提供了一些可极大简…
-
LSTM 的使用（Pytorch）
LSTM 参数 input_size:输入维数 hidden_size:输出维数 num_layers:LSTM层数，默认是1 bias:True 或者 False，决定是否使用bias, False则b_h=0. 默认为True batch_first:True 或者 False，因为nn.lstm()接受的数据输入是(序列长度，batch，输入维数)，这…
-
训练一个图像分类器demo in PyTorch【学习笔记】
【学习源】Tutorials > Deep Learning with PyTorch: A 60 Minute Blitz > Training a Classifier 本文相当于对上面链接教程中自认为有用部分进行的截取、翻译和再注释。便于日后复习、修正和补充。 边写边查资料的过程中猛然发现这居然有中文文档……不过中文文档也是志愿者翻译的，…
-
Pytorch实现波阻抗反演
Pytorch实现波阻抗反演 1 引言 地震波阻抗反演是在勘探与开发期间进行储层预测的一项关键技术。地震波阻抗反演可消除子波影响，仅留下反射系数，再通过反射系数计算出能表征地层物性变化的物理参数。常用的有道积分、广义线性反演、稀疏脉冲反演、模拟退火反演等技术。 随着勘探与开发的深入，研究的地质目标已经从大套厚层砂体转向薄层砂体，而利用常规波阻抗反演方法刻画薄…
-
从零搭建Pytorch模型教程（四）编写训练过程–参数解析
前言 训练过程主要是指编写train.py文件，其中包括参数的解析、训练日志的配置、设置随机数种子、classdataset的初始化、网络的初始化、学习率的设置、损失函数的设置、优化方式的设置、tensorboard的配置、训练过程的搭建等。由于篇幅问题，这些内容将分成多篇文章来写。本文介绍参数解析的两种方式。 欢迎关注公众号CV技术指南，专注于计算机…
-
实践torch.fx第一篇——基于Pytorch的模型优化量化神器
第一篇——什么是torch.fx 今天聊一下比较重要的torch.fx，也趁着这次机会把之前的torch.fx笔记整理下，笔记大概拆成三份，分别对应三篇： 什么是torch.fx 基于torch.fx做量化 基于torch.fx量化部署到TensorRT 本文对应第一篇，主要介绍torch.fx和基本使用方法。废话不多说，直接开始吧！ 什么是Torch.FX…
-
【深度学习 论文篇 03-2】Pytorch搭建SSD模型踩坑集锦
论文地址：https://arxiv.org/abs/1512.02325 源码地址：http://github.com/amdegroot/ssd.pytorch 环境1：torch1.9.0+CPU 环境2：torch1.8.1+cu102、torchvision0.9.1+cu102 1. StopIteration。Batch_size设置32，…
