长短期记忆网络(Long Short-Term Memory, LSTM)是一种递归神经网络(Recurrent Neural Network, RNN)的变体,专门用于处理和预测序列数据。它通过引入门控机制和记忆细胞,能够更好地捕捉序列中的长期依赖关系,并解决传统RNN中的梯度消失或爆炸问题。
2024-04-09 16:35:28 2KB pytorch pytorch lstm NLP
1
pytorch 一维信号处理,故障诊断,西储大学数据集,tsne,混肴矩阵等图片显示
2024-04-09 14:34:04 375.03MB pytorch pytorch
1
使用ESRGAN增强压缩图像的主观质量 0.背景 PyTorch实施压缩图像的主观质量增强。 注意:网络结构,数据集和训练方法与原始论文不同。 随时联系: ryanxingql@gmail.com 。 1.预先要求 1.1。 环境 乌本图20.04 LTS CUDA 10.1 烟火1.6 软件包:TQDM,LMDB,PYYAML,OPENCV-PYTHON,SCIKIT-IMAGE,TENSORBOARDX,LPIPS 假设您已经安装了CUDA 10.1,则: git clone --depth=1 https://github.com/RyanXingQL/SubjectiveQE-ESRGAN cd SubjectiveQE-ESRGAN/ conda create -n esrgan python=3.7 -y conda activate esrgan pytho
2024-04-09 14:07:30 40KB Python
1
pytorch yolov3 目标检测 yolov3-tiny.conv.15 yolov3 yolov3-tiny.conv.15 权重文件
2024-04-09 11:53:31 27.39MB yolov3-tiny.conv pytorch yolov3-t 目标检测
1
DETR(DEtection TRansformer)是一种基于Transformer架构的端到端目标检测模型,其主要流程包括: 1. 特征提取:通过卷积神经网络提取输入图像的特征。 2. Transformer编码器:将特征图输入Transformer编码器,利用自注意力机制和全连接层获取位置的上下文信息。 3. 对象查询:引入特殊的“对象”查询向量,指导模型在每个位置关注的对象类别。 4. 解码器:将Transformer编码器的输出作为解码器的输入,通过多层自注意力计算和全连接层计算生成每个位置的对象特征。 5. 对象匹配:将对象特征与所有可能的目标类别进行匹配,产生候选框和得分。 6. 位置预测:为每个候选框产生精确的位置预测。 DETR简化了目标检测流程,无需使用锚框或非极大值抑制,直接输出目标检测结果
2024-04-08 14:47:10 942KB transformer 目标检测 DETR pytorch
1
深度学习框架,gpu版本的pytorch,在python3.5+ cuda10.0 + cudnn7.6+pytorch1.2.0 gpu_torcvision0.4.0
2024-04-07 22:23:24 714.94MB 深度学习,pytorch gpu
1
在colab环境下实现,如果想使用cpu的话 自己换一下device的代码就可以了
2024-04-07 17:22:27 1.05MB pytorch pytorch lstm
1
使用LSTM实现C-MAPSS数据集里面的剩余寿命预测(Pytorch) 每轮训练后测试集误差 score:445.4610 334.5140 358.6489 365.9250 331.4520 283.3463 460.4766 314.7196 325.5950 452.3746 RMSE:16.3614 14.8254 14.9796 15.5157 14.7853 14.2053 16.2834 14.6757 14.7481 15.8802 由实验结果可知,MS-BLSTM 的预测误差均为最低水平,并且实际训练过程中收敛速度较快,涡扇发动机接近损坏时预测准确率较高。与传统机器学习方法相比,深度学习模型如CNN 和 LSTM的预测误差相对较小。而本文所提的 MS-BLSTM 混合深度学习预测模型进一步提高了 RUL 预测精度,,这得益于 MS-BLSTM 混合模型有效利用了时间段内传感器测量值的均值和方差与RUL的相关性,并使用 BLSTM学习历史数据和未来数据的长程依赖。本文所提的 MS-BLSTM 剩余使用寿命预测模型预测精度高,可有力支撑涡扇发动机的健康管理与运维决策。
2024-04-03 15:06:07 13.62MB pytorch pytorch lstm 数据集
1
PyTorch中的MeshCNN SIGGRAPH 2019 MeshCNN是用于3D三角形网格的通用深度神经网络,可用于诸如3D形状分类或分割之类的任务。 该框架包括直接应用于网格边缘的卷积,池化和解池层。 该代码由和在支持下编写。 入门 安装 克隆此仓库: git clone https://github.com/ranahanocka/MeshCNN.git cd MeshCNN 安装依赖项: 1.2版。 可选: 用于训练图。 通过新的conda环境conda env create -f environment.yml (创建一个名为meshcnn的环境) SHREC上的3D形状分类 下载数据集 bash ./scripts/shrec/get_data.sh 运行训练(如果使用conda env首先激活env,例如source activate meshcnn ) bash ./scripts/shrec/train.sh 要查看训练损失图,请在另一个终端中运行tensorboard --logdir runs并单击 。 运行测试并导出中间池网格: bas
2024-04-02 16:20:14 3.54MB machine-learning computer-graphics pytorch mesh
1
机器学习多层感知器实践完整源代码,MLP识别MNIST手写数字数据集(Pytorch)
2024-03-29 16:35:48 22.52MB pytorch 数据集 MNIST 机器学习
1