- 博客(13)
- 收藏
- 关注
原创 R2 LSTM-火灾温度预测
输入:当前时刻网络输出x_t,上一时刻LSTM输出h_t-1,上一时刻记忆单元c_t-1输出:当前时刻LSTM输出h_t,隐藏状态向量h_t, 记忆单元c_t。
2024-03-20 11:35:32 670 1
原创 P9 YOLOv5-Backbone模块实现
学习了YOLOv5的backbone模块的搭建,详细知识见模型建立部分。增加了模型的复杂度后,在前20轮训练就得到了准确率的提升,经过50轮训练得到了更好的性能。
2024-03-01 11:37:03 806 1
原创 P7 VGG16识别咖啡豆
进行VGG-16的手动搭建,熟悉VGG-16的网络结构。使用torchsummary进行模型信息的查看,包括各层的OutputShape和参数量,以及总体的参数量和大小。模型修改:增加BN层加速收敛,防止梯度爆炸和梯度消失,防止过拟合;增加dropout层,增加模型泛化能力,防止过拟合;使用nn.AdaptiveAvgPool2d——二维自适应平均池化运算,进行全局平均池化替代全连接层,可以抑制过拟合,减少参数量和计算量。修改后的模型实现了轻量化和准确率提升。
2024-02-21 17:58:46 576
原创 P1 mnist手写数字识别
本周内容主要是使用pytorch框架实现了之前在TensorFlow上完成过的mnist手写数字识别。两个架构在整体的思路上基本一致,但具体的实现函数较为不同。在后续的学习中可以结合对TensorFlow的回顾来对比两个框架。
2023-12-20 17:05:34 767
原创 T10 数据增强学习记录
这是大家可以自由发挥的一个地方# 随机改变图像对比度将aug_img应用于图像预处理preprocess_image:(重新加载了一次数据集)data_dir,seed=12,data_dir,seed=12,# 归一化处理train_ds = prepare(train_ds) #加入旋转])epochs=20train_ds,Epoch 1/20Epoch 2/20Epoch 3/20Epoch 4/20Epoch 5/20Epoch 6/20。
2023-11-29 16:48:59 975 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人