![](https://img-blog.csdnimg.cn/2019092715111047.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
pytorch
无悔丶
人生苦短
展开
-
详解Centernet训练自己的数据集(win10 + cuda10 + pytorch1.0.1)
centernet训练自己的数据集(win10 + cuda10 + pytorch1.0.1)Centernet避免本文太长,先分几部分分别介绍1、centernet论文和理论2、搭建centernet环境(win10+pytorch1.0.1)3、训练自己的数据集(报错问题)AttributeError: Can't pickle local object 'get_dataset...原创 2020-01-10 17:28:48 · 10548 阅读 · 50 评论 -
Pytorch实现RNN循环神经网络(实例)以及比较不同优化器对应的loss曲线
最近在b站上学习了几个关于pytorch框架视频,特地做了下笔记。一、线性回归最简单的一个线性回归问题,拟合一个二次函数,代码很简单,就不赘述了,直接贴上代码:import torchfrom torch.autograd import Variableimport torch.nn.functional as Fimport matplotlib.pyplot as pltx ...原创 2019-11-07 14:36:46 · 2901 阅读 · 0 评论 -
利用神经网络对图像分类(对29种不同云层图像分类-pytorch)附代码
一、训练集参加了一个气象比赛,记录一下训练过程数据集是在比赛官网上下载的;点这里说明一下,数据集大部分应该是比赛主办方在网上爬的,所以下载不了数据集也没关系,自己写个爬虫程序下载一下一样的。二、训练过程我用的是pytorch框架写的,主要是用了个残差网络进行训练,具体代码如下:# -*- encoding:utf-8 -*-import torchimport osfrom t...原创 2019-11-06 14:38:15 · 3295 阅读 · 5 评论