神兽乌鸦

Pytorch之TorchScript和trace保存网路的区别

本教程介绍如何把detectron2转为onnx

ONNX和NCNN到底是什么？

图像分类和目标检测有什么不同之处？

训练数据训练代码#导入库import torchimport torch.nn as nn#处理数据from torchtext.legacy import dataimport torch.optim as optimimport numpy as np#中文分词处理工具import jiebaimport reBATCH_SIZE = 2#产生同样的结果SEED = 2019torch.manual_seed(SEED)#Cuda 算法torch.backen

本篇文章基于《Colab在线解压Google Driver上的zip压缩文件》，所以有不懂的同学建议一并看完上一篇文章的内容，再来学习本篇文章。做深度学习需要大量的数据来进行训练，Kaggle上有很多现成的数据，本篇文章的目的就是把数据下载过来，然后训练。一、你需要再Kaggle上创建一个账号，然后按照下图去创建一个token点击Create New API Token按钮后会提示下载kaggle.json文件，保存到本地。二、把下载的kaggle.json上传到Google云盘存放的目录你自

今天才发现可以用Google的Colaboratory可以训练pytorch神经网络，真的太高兴了，这样就可以解决我渣渣的电脑没有GPU，无法训练大型神经网络的问题了。但是由于需要上传本地训练数据，很不方便，所以我就打算把本地数据打zip包上传到Google云盘，然后再在colab上用命令行下载解压过来训练，找了半天才找到办法，这里记录一下吧，主要代码如下：# 首先安装googledrivedownloaderpip install googledrivedownloader#接着下载解压from

本篇博文纯粹是作为一个记录，不会讲得太清楚，但是本人亲自运行验证，损失率达到了0.001，还比较满意，毕竟这是我的第一个机器识别程序。先看看项目目录：其中net.pkl和net_params.pkl是训练后保存的模型数据，data文件夹下放有训练用的数据和测试数据训练的数据我是从这里下载的训练数据下载接下来是训练用到的代码：#encoding=utf-8import torchi...

上一篇文章《pytorch实现性别识别，男人女人分类器》我们用pytorch实现了性别识别神经网络的训练和测试，这篇文章我们来介绍如何把训练好的模型迁移到Android设备上。一、Android上引入pytorch在app module下的build.gradle上加上 implementation 'org.pytorch:pytorch_android:1.3.0' im...