DeepSparkHub

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

模型列表

注：当前列表中模型主要基于PyTorch框架，基于PaddlePaddle、TensorFlow、MindSpore等框架的模型在逐步添加中。

Computer Vision
Multimodal
GNN
Natural Language Processing(NLP)
Recommendation
- Collaborative Filtering
- CTR
Speech
- Speech Recognition
- Speech Synthesis
3D Reconstruction

Computer Vision

Classification

模型名称	框架	数据集
ACmix	PyTorch	ImageNet
AlexNet	PyTorch	ImageNet
CBAM	PyTorch	ImageNet
ConvNext	PyTorch	ImageNet
CspDarknet53	PyTorch	ImageNet
DenseNet	PyTorch	ImageNet
DPN92	PyTorch	ImageNet
DPN107	PyTorch	ImageNet
ECA_MobileNet_V2	PyTorch	ImageNet
ECA_RESNET152	PyTorch	ImageNet
Efficientb4	PyTorch	ImageNet
FasterNet	PyTorch	ImageNet
googlenet	PyTorch	ImageNet
googlenet	PaddlePaddle	ImageNet
InceptionV3	PyTorch	ImageNet
InceptionV4	PyTorch	ImageNet
lenet	PyTorch	ImageNet
MobileNetV2	PyTorch	ImageNet
MobileNetV3	PyTorch	ImageNet
MobileNetV3	PaddlePaddle	ImageNet
RepVGG	PyTorch	ImageNet
RepVGG	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	PyTorch	ImageNet
ResNeXt101_32x8d	PyTorch	ImageNet
SEResNeXt	PyTorch	ImageNet
ShuffleNetV2	PyTorch	ImageNet
SqueezeNet	PyTorch	ImageNet
Swin Transformer	PyTorch	ImageNet
Swin Transformer	PaddlePaddle	ImageNet
VGG16	PyTorch	ImageNet
VGG16	PaddlePaddle	ImageNet
Wave-MLP	PyTorch	ImageNet
Wide_ResNet101_2	PyTorch	ImageNet
Xception	PyTorch	ImageNet

Face Detection

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

Face Recognition

模型名称	框架	数据集
ArcFace	PyTorch	CASIA-WebFaces&LFW
CosFace	PyTorch	CASIA-WebFaces&LFW
FaceNet	PyTorch	CASIA-WebFaces&LFW

Instance Segmentation

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

Image Generation

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

Knowledge Distillation

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

Network Pruning

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

Object Detection

模型名称	框架	数据集
ATSS	PyTorch (MMDetection)	COCO
AutoAssign	PyTorch	COCO
Cascade R-CNN	PyTorch (MMDetection)	COCO
CenterNet	PyTorch	COCO
CenterNet	PaddlePaddle	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
Mask R-CNN	PyTorch	COCO
Mask R-CNN	PaddlePaddle	COCO
PP-YOLOE	PaddlePaddle	COCO
PVANet	PyTorch	COCO
RepPoints	PyTorch (MMDetection)	COCO
RetinaNet	PyTorch	COCO
RetinaNet	PaddlePaddle	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	PyTorch	COCO
YOLOv6	PyTorch	COCO
YOLOv7	PyTorch	COCO
YOLOv8	PyTorch	COCO

3D Object Detection

模型名称	框架	数据集
BEVFormer	PyTorch	nuScenes&CAN bus
PointNet++	PyTorch	S3DIS
PointPillars	PyTorch	KITTI

OCR

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

Point Cloud

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

Pose Estimation

模型名称	框架	数据集
AlphaPose	PyTorch	COCO
HRNet	PyTorch	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
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
DenseASPP	PyTorch	COCO
DFANet	PyTorch	COCO
dnlnet	PaddlePaddle	Cityscapes
DUNet	PyTorch	COCO
EncNet	PyTorch	COCO
ENet	PyTorch	COCO
ERFNet	PyTorch	COCO
ESPNet	PyTorch	COCO
FastSCNN	PyTorch	COCO
FCN	PyTorch	COCO
FPENet	PyTorch	COCO
GCNet	PyTorch	Cityscapes
HardNet	PyTorch	COCO
ICNet	PyTorch	COCO
LedNet	PyTorch	COCO
LinkNet	PyTorch	COCO
OCNet	PyTorch	COCO
OCRNet	PyTorch	Cityscapes
PSANet	PyTorch	COCO
RefineNet	PyTorch	COCO
SegNet	PyTorch	COCO
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

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

Traffic Forecast

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

Multimodal

模型名称	框架	数据集
CLIP	PyTorch	CIFAR100
L-Verse	PyTorch	ImageNet
Stable Diffusion	PyTorch	pokemon-images

GNN

Text Classification

模型名称	框架	数据集
GCN	MindSpore	Cora&Citeseer

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
RoBERTa	PyTorch (Fairseq)	RTE

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

CTR

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

Speech

Speech Recognition

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

Speech Synthesis

模型名称	框架	数据集
Tacotron2	PyTorch	LJSpeech
vqmivc	PyTorch	VCTK-Corpus
waveglow	PyTorch	LJSpeech

3D Reconstruction

模型名称	框架	数据集
ngp-nerf	PyTorch	fox

容器镜像构建方式

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

社区

治理

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

交流

请联系 contact@deepspark.org.cn。

贡献

请参见 DeepSparkHub Contributing Guidelines。

许可证

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

simmoncn/DeepSparkHub