Skip to content
/ DeepSparkHub Public

DeepSparkHub selects hundreds of application algorithms and models, covering various fields of AI and general-purpose computing, to support the mainstream intelligent computing scenarios.

License

Apache-2.0 license
58 stars 11 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

Deep-Spark/DeepSparkHub

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

866 Commits
.gitee
.gitee
 
 
3d-reconstruction/hashnerf/pytorch
3d-reconstruction/hashnerf/pytorch
 
 
cv
cv
 
 
docker/Iluvatar
docker/Iluvatar
 
 
docs
docs
 
 
gnn
gnn
 
 
hpc/molecular_dynamics/water_se_e2_a/tensorflow
hpc/molecular_dynamics/water_se_e2_a/tensorflow
 
 
methodology/kolmogorov_arnold_networks/kan/pytorch
methodology/kolmogorov_arnold_networks/kan/pytorch
 
 
multimodal
multimodal
 
 
nlp
nlp
 
 
recommendation
recommendation
 
 
reinforcement_learning/q-learning-networks/dqn/paddlepaddle
reinforcement_learning/q-learning-networks/dqn/paddlepaddle
 
 
speech
speech
 
 
toolbox
toolbox
 
 
.gitignore
.gitignore
 
 
CONTRIBUTING.md
CONTRIBUTING.md
 
 
LICENSE
LICENSE
 
 
README.md
README.md
 
 
RELEASE.md
RELEASE.md
 
 

Repository files navigation

DeepSparkHub

DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领域，支持主流市场智能计算场景，包括智慧城市、数字个人、医疗、教育、通信、能源等多个领域。

模型列表

Computer Vision

Classification

模型名称 框架 数据集
ACmix PyTorch ImageNet
ACNet PyTorch ImageNet
AlexNet PyTorch ImageNet
AlexNet TensorFlow ImageNet
BYOL PyTorch ImageNet
CBAM PyTorch ImageNet
ConvNext PyTorch ImageNet
CspDarknet53 PyTorch ImageNet
DenseNet PaddlePaddle ImageNet
DenseNet PyTorch ImageNet
DPN92 PyTorch ImageNet
DPN107 PyTorch ImageNet
ECA-MobileNetV2 PyTorch ImageNet
ECA-ResNet152 PyTorch ImageNet
EfficientNetB0 PaddlePaddle ImageNet
EfficientNetB4 PyTorch ImageNet
FasterNet PyTorch ImageNet
GoogLeNet PyTorch ImageNet
GoogLeNet PaddlePaddle ImageNet
InceptionV3 MindSpore ImageNet
InceptionV3 PyTorch ImageNet
InceptionV3 TensorFlow ImageNet
InceptionV4 PyTorch ImageNet
InternImage PyTorch ImageNet
LeNet PyTorch ImageNet
MobileNetV2 PyTorch ImageNet
MobileNetV3 MindSpore ImageNet
MobileNetV3 PyTorch ImageNet
MobileNetV3 PaddlePaddle ImageNet
MobileNetV3_Large1.0 PaddlePaddle ImageNet
MobileOne PyTorch ImageNet
MoCoV2 PyTorch ImageNet
PP-LCNet PaddlePaddle ImageNet
RepMLP PyTorch ImageNet
RepVGG PyTorch ImageNet
RepVGG PaddlePaddle ImageNet
RepViT PyTorch ImageNet
Res2Net50_14w_8s PaddlePaddle ImageNet
ResNeSt14 PyTorch ImageNet
ResNeSt50 PyTorch ImageNet
ResNeSt50 PaddlePaddle ImageNet
ResNeSt101 PyTorch ImageNet
ResNeSt269 PyTorch ImageNet
ResNet18 PyTorch ImageNet
ResNet50 PyTorch ImageNet
ResNet50 PaddlePaddle ImageNet
ResNet50 TensorFlow ImageNet
ResNet101 PyTorch ImageNet
ResNet152 PyTorch ImageNet
ResNeXt50_32x4d MindSpore ImageNet
ResNeXt50_32x4d PyTorch ImageNet
ResNeXt101_32x8d PyTorch ImageNet
SE_ResNet50_vd PaddlePaddle ImageNet
SEResNeXt PyTorch ImageNet
ShuffleNetV2 PaddlePaddle ImageNet
ShuffleNetV2 PyTorch ImageNet
SqueezeNet PyTorch ImageNet
Swin Transformer PaddlePaddle ImageNet
Swin Transformer PyTorch ImageNet
VGG16 PaddlePaddle ImageNet
VGG16 PyTorch ImageNet
VGG16 TensorFlow ImageNet
Wave-MLP PyTorch ImageNet
Wide_ResNet101_2 PyTorch ImageNet
Xception PaddlePaddle ImageNet
Xception PyTorch ImageNet

Face Detection

模型名称 框架 数据集
RetinaFace PyTorch WiderFace

Face Recognition

模型名称 框架 数据集
ArcFace PyTorch CASIA-WebFaces&LFW
BlazeFace PaddlePaddle WIDER-FACE
CosFace PyTorch CASIA-WebFaces&LFW
FaceNet PyTorch CASIA-WebFaces&LFW
FaceNet TensorFlow CASIA-WebFaces&LFW

Instance Segmentation

模型名称 框架 数据集
SOLO PyTorch COCO
SOLOv2 PaddlePaddle COCO
SOLOv2 PyTorch COCO
YOLACT++ PyTorch COCO

Image Generation

模型名称 框架 数据集
DCGAN MindSpore ImageNet
Pix2Pix PaddlePaddle facades

Knowledge Distillation

模型名称 框架 数据集
CWD PyTorch Cityscapes
RKD PyTorch CUB-200-2011
WSLD PyTorch ImageNet

Network Pruning

模型名称 框架 数据集
Network Slimming PyTorch CIFAR-10/100

Object Detection

模型名称 框架 数据集
ATSS PyTorch (MMDetection) COCO
AutoAssign PyTorch COCO
Cascade R-CNN PyTorch (MMDetection) COCO
CenterMask2 PyTorch COCO
CenterNet PyTorch COCO
CenterNet PaddlePaddle COCO
Co-DETR PyTorch COCO
CornerNet PyTorch (MMDetection) COCO
DCNV2 PyTorch (MMDetection) COCO
DeepSORT PyTorch Market-1501
DETR PaddlePaddle COCO
Faster R-CNN PyTorch COCO
FCOS PaddlePaddle COCO
FCOS PyTorch COCO
Mamba-YOLO PyTorch COCO
Mask R-CNN PyTorch COCO
Mask R-CNN PaddlePaddle COCO
OC_SORT PaddlePaddle MOT17
Oriented RepPoints PyTorch DOTA
PP-PicoDet PaddlePaddle COCO
PP-YOLOE PaddlePaddle COCO
PP-YOLOE+ PaddlePaddle COCO
PVANet PyTorch COCO
RepPoints PyTorch (MMDetection) COCO
RetinaNet PyTorch COCO
RetinaNet PaddlePaddle COCO
RT-DETR PyTorch COCO
RTMDet PyTorch COCO
SSD PyTorch COCO
SSD PaddlePaddle COCO
SSD TensorFlow VOC
SSD MindSpore COCO
YOLOF PyTorch COCO
YOLOv3 PyTorch COCO
YOLOv3 PaddlePaddle COCO
YOLOv3 TensorFlow VOC
YOLOv5 PaddlePaddle COCO
YOLOv5 PyTorch COCO
YOLOv6 PyTorch COCO
YOLOv7 PyTorch COCO
YOLOv8 PyTorch COCO
YOLOv9 PyTorch COCO
YOLOv10 PyTorch COCO

3D Object Detection

模型名称 框架 数据集
BEVFormer PyTorch nuScenes&CAN bus
CenterPoint PyTorch nuScenes
PAConv PyTorch S3DIS
Part-A2-Anchor PyTorch KITTI
Part-A2-Free PyTorch KITTI
PointNet++ PyTorch S3DIS
PointPillars PyTorch KITTI
PointRCNN PyTorch KITTI
PointRCNN-IoU PyTorch KITTI
SECOND PyTorch KITTI
SECOND-IoU PyTorch KITTI

OCR

模型名称 框架 数据集
CRNN MindSpore OCR_Recog
CRNN PaddlePaddle LMDB
DBNet PyTorch ICDAR2015
DBNet++ PaddlePaddle ICDAR2015
DBNet++ PyTorch ICDAR2015
PP-OCR-DB PaddlePaddle ICDAR2015
PP-OCR-EAST PaddlePaddle ICDAR2015
PSE PaddlePaddle OCR_Recog
SAR PyTorch OCR_Recog
SAST PaddlePaddle ICDAR2015
SATRN PyTorch OCR_Recog

Point Cloud

模型名称 框架 数据集
Point-BERT PyTorch ShapeNet55 & processed ModelNet

Pose Estimation

模型名称 框架 数据集
AlphaPose PyTorch COCO
HRNet PyTorch COCO
HRNet-W32 PaddlePaddle COCO
OpenPose MindSpore COCO

Self-Supervised Learning

模型名称 框架 数据集
MAE PyTorch ImageNet

Semantic Segmentation

模型名称 框架 数据集
3D-UNet PyTorch kits19
APCNet PyTorch Cityscapes
Attention U-net PyTorch Cityscapes
BiSeNet PyTorch COCO
BiSeNetV2 PaddlePaddle Cityscapes
BiSeNetV2 PyTorch Cityscapes
CGNet PyTorch COCO
ContextNet PyTorch COCO
DabNet PyTorch COCO
DANet PyTorch COCO
DDRnet PyTorch Cityscapes
DeepLabV3 PyTorch COCO
DeepLabV3 PaddlePaddle Cityscapes
DeepLabV3 MindSpore VOC
DeepLabV3+ PaddlePaddle Cityscapes
DeepLabV3+ TensorFlow Cityscapes
DenseASPP PyTorch COCO
DFANet PyTorch COCO
DNLNet PaddlePaddle Cityscapes
DUNet PyTorch COCO
EncNet PyTorch COCO
ENet PyTorch COCO
ERFNet PyTorch COCO
ESPNet PyTorch COCO
FastFCN PyTorch ADE20K
FastSCNN PyTorch COCO
FCN PyTorch COCO
FPENet PyTorch COCO
GCNet PyTorch Cityscapes
HardNet PyTorch COCO
ICNet PyTorch COCO
LedNet PyTorch COCO
LinkNet PyTorch COCO
Mask2Former PyTorch Cityscapes
MobileSeg PaddlePaddle Cityscapes
OCNet PyTorch COCO
OCRNet PaddlePaddle Cityscapes
OCRNet PyTorch Cityscapes
PP-HumanSegV1 PaddlePaddle PP-HumanSeg14K
PP-HumanSegV2 PaddlePaddle PP-HumanSeg14K
PP-LiteSeg PaddlePaddle Cityscapes
PSANet PyTorch COCO
RefineNet PyTorch COCO
SegNet PyTorch COCO
STDC PaddlePaddle Cityscapes
STDC PyTorch Cityscapes
UNet PyTorch COCO
UNet PaddlePaddle Cityscapes
UNet++ PyTorch DRIVE
VNet TensorFlow Hippocampus

Super Resolution

模型名称 框架 数据集
basicVSR++ PyTorch REDS
basicVSR PyTorch REDS
ESRGAN PyTorch DIV2K
LIIF PyTorch DIV2K
RealBasicVSR PyTorch REDS
TTSR PyTorch CUFED
TTVSR PyTorch REDS

Tracking

模型名称 框架 数据集
ByteTrack PaddlePaddle MOT17
FairMOT PyTorch MOT17

Traffic Forecast

模型名称 框架 数据集
Graph WaveNet PyTorch METR-LA & PEMS-BAY

GNN

Graph Attention

模型名称 框架 数据集
GAT PaddlePaddle CORA

Node Classification

模型名称 框架 数据集
GraphSAGE PaddlePaddle Reddit

Text Classification

模型名称 框架 数据集
GCN MindSpore CORA & Citeseer
GCN PaddlePaddle CORA & PubMed & Citeseer

HPC

Molecular Dynamics

模型名称 框架 数据集
Water/se_e2_a TensorFlow (DeePMD-kit) data_water

Methodology

Kolmogorov-Arnold Networks

模型名称 框架 数据集
KAN PyTorch -

Multimodal

模型名称 框架 数据集
BLIP PyTorch COCO
CLIP PyTorch CIFAR100
ControlNet PyTorch Fill50K
DDPM PyTorch CIFAR-10
LLaVA 1.5 PyTorch LLaVA-Pretrain
L-Verse PyTorch ImageNet
Stable Diffusion 1.4 PyTorch pokemon-images
Stable Diffusion 1.5 PyTorch pokemon-images
Stable Diffusion 2.1 PyTorch pokemon-images
Stable Diffusion 3 PyTorch dog-example
Stable Diffusion XL PyTorch pokemon-images

NLP

Cloze Test

模型名称 框架 数据集
GLM PyTorch GLMForMultiTokenCloze

Dialogue Generation

模型名称 框架 数据集
CPM PyTorch STC

Language Modeling

模型名称 框架 数据集
BART PyTorch (Fairseq) RTE
BERT NER PyTorch CoNLL-2003
BERT Pretraining PyTorch MLCommon Wikipedia (2048_shards_uncompressed)
BERT Pretraining PaddlePaddle MNLI
BERT Pretraining TensorFlow MNLI
BERT Pretraining MindSpore SQuAD
BERT Text Classification PyTorch GLUE
BERT Text Summerization PyTorch cnn_dailymail
BERT Question Answering PyTorch SQuAD
GPT2-Medium-EN PaddlePaddle SST-2
RoBERTa PyTorch (Fairseq) RTE
XLNet PaddlePaddle SST-2

Large Language Model (LLM)

模型名称 框架 工具箱 数据集/权重
Aquila2-34B PyTorch Megatron-DeepSpeed Bookcorpus
Baichuan2-7B PyTorch DeepSpeed baichuan2-7b-base
Bloom-7B1 PyTorch Firefly school_math_0.25M & bloom-7b1
ChatGLM-6B PyTorch DeepSpeed ADGEN & chatglm-6b
ChatGLM2-6B SFT PyTorch DeepSpeed ADGEN & chatglm2-6b
ChatGLM3-6B PyTorch DeepSpeed ADGEN & chatglm3-6b
DeepSeekMoE 7B PyTorch ColossalAI deepseek-moe-16b-base
Llama-7B PyTorch Colossal-AI llama-7b-hf
Llama2-7B PyTorch Megatron-DeepSpeed Bookcorpus
Llama2-7B Reward Model Finetuning PyTorch DeepSpeed Dahoas/rm-static
Llama2-7B RLHF PyTorch Megatron-DeepSpeed llama2-7b&tiny-llama
Llama2-7B SFT PyTorch Megatron-DeepSpeed GPT Small-117M
Llama2-13B PyTorch Megatron-DeepSpeed Bookcorpus
Llama2-34B PyTorch Megatron-DeepSpeed Bookcorpus
Llama3-8B PyTorch Megatron-DeepSpeed Bookcorpus
Llama3-8B SFT PyTorch ColossalAI school_math_0.25M
Mamba-2 PyTorch Megatron-LM GPT Small-117M
Mixtral 8x7B PyTorch Megatron-LM GPT Small-117M
QWen-7B PyTorch Firefly qwen-7b
QWen1.5-7B PyTorch Firefly school_math
QWen1.5-14B PyTorch Firefly school_math
Qwen2.5-7B SFT PyTorch LLaMA-Factory qwen2.5-7b

Text Correction

模型名称 框架 数据集
Ernie PaddlePaddle corpus

Translation

模型名称 框架 数据集
Convolutional PyTorch (Fairseq) WMT14
T5 PyTorch wmt14-en-de-pre-processed
Transformer PaddlePaddle wmt14-en-de-pre-processed
Transformer PyTorch (Fairseq) IWSLT14

Recommendation

Collaborative Filtering

模型名称 框架 数据集
NCF PyTorch movielens

Click Through Rate

模型名称 框架 数据集
DLRM PyTorch Criteo_Terabyte
DLRM PaddlePaddle Criteo_Terabyte
FFM PaddlePaddle Criteo_Terabyte
DeepFM PaddlePaddle Criteo_Terabyte
Wide&Deep PaddlePaddle Criteo_Terabyte
xDeepFM PaddlePaddle Criteo_Terabyte

Reinforcement Learning

模型名称 框架 数据集
DQN PaddlePaddle CartPole-v0

Speech

Speech Recognition

模型名称 框架 数据集
Conformer PyTorch (WeNet) AISHELL
Efficient Conformer v2 PyTorch (WeNet) AISHELL
PP-ASR-Conformer PaddlePaddle AISHELL
RNN-T PyTorch LJSpeech
Transformer PyTorch (WeNet) AISHELL
U2++ Conformer PyTorch (WeNet) AISHELL
Unified Conformer PyTorch (WeNet) AISHELL

Speech Synthesis

模型名称 框架 数据集
PP-TTS-FastSpeech2 PaddlePaddle CSMSC
PP-TTS-HiFiGAN PaddlePaddle CSMSC
Tacotron2 PyTorch LJSpeech
VQMIVC PyTorch VCTK-Corpus
WaveGlow PyTorch LJSpeech

3D Reconstruction

模型名称 框架 数据集
HashNeRF PyTorch fox

容器镜像构建方式

社区用户可参考容器镜像构建说明在本地构建出能够运行DeepSparkHub仓库中模型的容器镜像。

社区

治理

请参见 DeepSpark Code of Conduct on Gitee or on GitHub。

交流

请联系 [email protected]。

贡献

请参见 DeepSparkHub Contributing Guidelines。

免责声明

DeepSparkHub仅提供公共数据集的下载和预处理脚本。这些数据集不属于DeepSparkHub，DeepSparkHub也不对其质量或维护负责。请确保您具有这些数据集的使用许可，基于这些数据集训练的模型仅可用于非商业研究和教育。

致数据集所有者：

如果不希望您的数据集公布在DeepSparkHub上或希望更新DeepSparkHub中属于您的数据集，请在Gitee或Github上提交issue，我们将按您的issue删除或更新。衷心感谢您对我们社区的支持和贡献。

许可证

本项目许可证遵循Apache-2.0。

About

DeepSparkHub selects hundreds of application algorithms and models, covering various fields of AI and general-purpose computing, to support the mainstream intelligent computing scenarios.

Resources

Readme

License

Apache-2.0 license
Activity
Custom properties

Stars

58 stars

Watchers

6 watching

Forks

11 forks
Report repository

Releases 9

DeepSparkHub 24.12 Release Latest
Dec 23, 2024
+ 8 releases

Contributors 9

Languages