半监督分层递归图神经网络用于城市范围内的停车位可用性预测 这是SHARE体系结构的Pytorch实现,如论文《。 如果您在研究中利用SHARE模型,请引用以下内容: @article{zhang2019semi, title={Semi-Supervised Hierarchical Recurrent Graph Neural Network for City-Wide Parking Availability Prediction}, author={Zhang, Weijia and Liu, Hao and Liu, Yanchi and Zhou, Jingbo and Xiong, Hui}, booktitle={Proceedings of the Thirty-Fourth AAAI Conference on Artificial Intelligen
1
nanodet source file and version for pytorch
2022-12-08 11:28:22 15.55MB 深度学习
1
YOLODet-PyTorch是端到端基于pytorch框架复现yolo最新算法的目标检测开发套件,旨在帮助开发者更快更好地完成检测模型的训练、精度速度优化到部署全流程。以模块化的设计实现了多种主流YOLO目标检测算法,并且提供了丰富的数据增强、网络组件、损失函数等模块。
2022-12-07 16:28:14 6.85MB 人工智能 AI-人工智能
1
本项目开发了PyTorch Adapter插件,用于昇腾适配PyTorch框架,为使用PyTorch框架的开发者提供昇腾AI处理器的超强算力。用户在准备相关环境进行基于PyTorch框架模型的开发、运行时,可以选择在服务器中手动编译相关模块。
2022-12-07 16:28:14 13.09MB 人工智能 PyTorch AI-人工智能
1
densenet121官方预训练模型,可以用于迁移学习训练
2022-12-07 15:14:05 28.93MB 迁移学习 人工智能 机器学习 pytorch
1
基于改进SSD算法(SE+特征融合)的苹果叶病虫害识别系统源码(pytorch框架)+改进前源码+病害数据集+项目说明.zip 主要改进点如下: 1、替换backbone为Resnet/MobileNet 2、添加一种更加轻量高效的特征融合方式 feature fusion module 3、添加注意力机制 (Squeeze-and-Excitation Module 和 Convolutional Block Attention Module) 4、添加一种解决正负样本不平衡的损失函数Focal Loss 附有苹果叶病害数据集,可训练模型
2022-12-07 12:27:48 90.31MB SSD 算法改进 注意力机制 SE模块
基于PyQt5+SRnet+SSDP网络实现图像隐写分析以及隐写去除_pytorch源码+项目说明.zip 项目主要任务为实现图像隐写分析以及隐写去除,其中隐写分析采用SRNet网络模型,隐写去除采用DDSP网络模型。 ​ 项目中有4个文件夹,分别为: 0.SRNet、1.GUI、2.DDSP、3.SRNet 其中0.SRNet为图像隐写分析,使用Jessica教授的官方源码,框架为tensorflow;1.GUI为隐写嵌入以及隐写分析可视化演示系统,由PyQ5实现;2.DDSP为图像隐写去除,pytorch实现;3.SRNet为图像隐写分析,pytorch实现。其中自己复现的SRNet网络模型其性能弱于官方代码。 本项目隐写分析中使用的隐写术为: S-UNIWARD、HUGO、WOW三种图像空域隐写算法,采用的隐写嵌入率为:0.4bpp、0.7bpp和1.0bpp三种。 为了更好的演示如何实现隐写嵌入和隐写分析,使用PyQt5编写了可视化界面,调用现有的隐写术和训练好的隐写分析模型进行操作。
2022-12-07 12:27:47 7.42MB PyQt5 SRnet SSDP 图像隐写分析
改进yolov5(多检测头+注意力机制+repvgg结构)pytorch源码+项目说明.zip 集成yolov5(v6.0), 注意力机制, 和repvgg结构 添加了多头检测代码,使用train_multiple_detection_head.py文件进行训练 添加了检测+关键点代码,使用train_key_point.py文件进行训练
Attention is all you need 解读及详细代码,每个模块都有注释,代码详细易懂。哈佛出品,值得阅读。Transformer最基本的结构,也是BERT的基本结构,NLP和推荐都需要用到。
2022-12-07 12:27:41 2.57MB attention Transformer BERT NLP
1
获取Fashion-MNIST训练集和读取数据 在介绍softmax回归的实现前我们先引入一个多类图像分类数据集。它将在后面的章节中被多次使用,以方便我们观察比较算法之间在模型精度和计算效率上的区别。图像分类数据集中最常用的是手写数字识别数据集MNIST[1]。但大部分模型在MNIST上的分类精度都超过了95%。为了更直观地观察算法之间的差异,我们将使用一个图像内容更加复杂的数据集Fashion-MNIST[2]。 我这里我们会使用torchvision包,它是服务于PyTorch深度学习框架的,主要用来构建计算机视觉模型。torchvision主要由以下几部分构成: torchvision.
2022-12-06 19:01:51 74KB c max OR
1