- 博客(13)
- 收藏
- 关注
RSS订阅原创 R2 LSTM-火灾温度预测
输入：当前时刻网络输出x_t，上一时刻LSTM输出h_t-1，上一时刻记忆单元c_t-1输出：当前时刻LSTM输出h_t，隐藏状态向量h_t， 记忆单元c_t。
2024-03-20 11:35:32
607
1
原创 P9 YOLOv5-Backbone模块实现
学习了YOLOv5的backbone模块的搭建，详细知识见模型建立部分。增加了模型的复杂度后，在前20轮训练就得到了准确率的提升，经过50轮训练得到了更好的性能。
2024-03-01 11:37:03
750
1
原创 P7 VGG16识别咖啡豆
进行VGG-16的手动搭建，熟悉VGG-16的网络结构。使用torchsummary进行模型信息的查看，包括各层的OutputShape和参数量，以及总体的参数量和大小。模型修改：增加BN层加速收敛，防止梯度爆炸和梯度消失，防止过拟合；增加dropout层,增加模型泛化能力，防止过拟合；使用nn.AdaptiveAvgPool2d——二维自适应平均池化运算，进行全局平均池化替代全连接层，可以抑制过拟合，减少参数量和计算量。修改后的模型实现了轻量化和准确率提升。
2024-02-21 17:58:46
560
原创 P1 mnist手写数字识别
本周内容主要是使用pytorch框架实现了之前在TensorFlow上完成过的mnist手写数字识别。两个架构在整体的思路上基本一致，但具体的实现函数较为不同。在后续的学习中可以结合对TensorFlow的回顾来对比两个框架。
2023-12-20 17:05:34
741
原创 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
933
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人