深度学习
赵 XiaoQin
Never Say Never
展开
-
NMS各种变体
目标检测之非极大值抑制(NMS)各种变体转载 2020-11-30 09:54:25 · 357 阅读 · 0 评论 -
AlexNet模型代码整理
1. 网络模型(论文是用两块gpu并行训练的,所以图中的模型是上下两部分的,第一层卷积核的个数是48+48=96)2. 模型参数3. 模型代码import torch.nn as nnimport torchclass AlexNet(nn.Module): def __init__(self, num_classes=1000, init_weights=False): super(AlexNet, self).__init__() .原创 2020-07-05 20:28:32 · 1150 阅读 · 0 评论 -
pytorch笔记8--optimizer
对比各种优化器的效果数据分布如下图:import torchimport torch.utils.data as Dataimport torch.nn.functional as Funcfrom matplotlib import pyplot as plttorch.manual_seed(1)#hyper parametersLR=0.01BATCH_SIZ...原创 2020-03-24 23:36:07 · 173 阅读 · 0 评论 -
深度学习笔记--Kaggle比赛之房价预测
#获取和读取数据import torchimport torch.nn as nnimport pandas as pd #处理数据import All_function as func #自定义包torch.set_default_tensor_type(torch.FloatTensor)train_data=pd.read_csv('Kaggle_...原创 2019-11-26 15:34:50 · 830 阅读 · 1 评论 -
深度学习笔记--Fashion_mnist+softmax的简单实现
1. 主文件import torchimport All_function as funcimport torch.nn as nn#获取数据batch_size=256train_iter,test_iter=func.load_data_fashion_minist(256)#定义和初始化模型class LinearNet(nn.Module): def __i...原创 2019-11-13 11:59:35 · 478 阅读 · 0 评论