DeepSparkHub

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

模型列表

LLM (Large Language Model)

Model	Framework	ToolBox	Dataset/Weight	IXUCA SDK
Aquila2-34B	PyTorch	Megatron-DeepSpeed	Bookcorpus	4.1.1
Baichuan2-7B	PyTorch	DeepSpeed	baichuan2-7b-base	4.1.1
Bloom-7B1	PyTorch	Firefly	school_math_0.25M	4.1.1
ChatGLM-6B	PyTorch	DeepSpeed	ADGEN & chatglm-6b	3.1.0
ChatGLM2-6B SFT	PyTorch	DeepSpeed	ADGEN & chatglm2-6b	4.1.1
ChatGLM3-6B	PyTorch	DeepSpeed	ADGEN & chatglm3-6b	4.1.1
DeepSeekMoE 7B	PyTorch	ColossalAI	deepseek-moe-16b-base	4.1.1
Llama-7B	PyTorch	ColossalAI	llama-7b-hf	3.1.0
Llama2-7B	PyTorch	Megatron-DeepSpeed	Bookcorpus	3.1.0
Llama2-7B RMF	PyTorch	DeepSpeed	Dahoas/rm-static	3.1.1
Llama2-7B RLHF	PyTorch	Megatron-DeepSpeed	llama2-7b&tiny-llama	4.1.1
Llama2-7B SFT	PyTorch	Megatron-DeepSpeed	GPT Small-117M	3.1.1
Llama2-13B	PyTorch	Megatron-DeepSpeed	Bookcorpus	4.1.1
Llama2-34B	PyTorch	Megatron-DeepSpeed	Bookcorpus	4.1.1
Llama3-8B	PyTorch	Megatron-DeepSpeed	Bookcorpus	4.1.1
Llama3-8B SFT	PyTorch	ColossalAI	school_math_0.25M	4.1.1
Mamba-2	PyTorch	Megatron-LM	GPT Small-117M	4.1.1
Mixtral 8x7B	PyTorch	Megatron-LM	GPT Small-117M	4.1.1
QWen-7B	PyTorch	Firefly	qwen-7b	4.1.1
QWen1.5-7B	PyTorch	Firefly	school_math	4.1.1
QWen1.5-14B	PyTorch	Firefly	school_math	4.1.1
Qwen2.5-7B SFT	PyTorch	LLaMA-Factory	qwen2.5-7b	4.1.1

Computer Vision

Classification

Model	Framework	Dataset	IXUCA SDK
ACmix	PyTorch	ImageNet	2.2.0
ACNet	PyTorch	ImageNet	3.1.0
AlexNet	PyTorch	ImageNet	2.2.0
AlexNet	TensorFlow	ImageNet	3.1.0
BYOL	PyTorch	ImageNet	3.1.0
CBAM	PyTorch	ImageNet	3.0.0
ConvNext	PyTorch	ImageNet	2.2.0
CspDarknet53	PyTorch	ImageNet	3.0.0
DenseNet	PaddlePaddle	ImageNet	3.1.0
DenseNet	PyTorch	ImageNet	2.2.0
DPN92	PyTorch	ImageNet	2.2.0
DPN107	PyTorch	ImageNet	2.2.0
ECA-MobileNetV2	PyTorch	ImageNet	2.2.0
ECA-ResNet152	PyTorch	ImageNet	2.2.0
EfficientNetB0	PaddlePaddle	ImageNet	3.1.0
EfficientNetB4	PyTorch	ImageNet	2.2.0
FasterNet	PyTorch	ImageNet	3.0.0
GoogLeNet	PyTorch	ImageNet	2.2.0
GoogLeNet	PaddlePaddle	ImageNet	2.3.0
InceptionV3	MindSpore	ImageNet	3.1.0
InceptionV3	PyTorch	ImageNet	2.2.0
InceptionV3	TensorFlow	ImageNet	3.1.0
InceptionV4	PyTorch	ImageNet	2.2.0
InternImage	PyTorch	ImageNet	3.1.0
LeNet	PyTorch	ImageNet	2.2.0
MobileNetV2	PyTorch	ImageNet	2.2.0
MobileNetV3	MindSpore	ImageNet	3.1.0
MobileNetV3	PyTorch	ImageNet	2.2.0
MobileNetV3	PaddlePaddle	ImageNet	2.3.0
MobileNetV3_Large1.0	PaddlePaddle	ImageNet	3.1.0
MobileOne	PyTorch	ImageNet	3.1.0
MoCoV2	PyTorch	ImageNet	3.1.0
PP-LCNet	PaddlePaddle	ImageNet	3.1.0
RepMLP	PyTorch	ImageNet	3.1.0
RepVGG	PyTorch	ImageNet	3.0.0
RepVGG	PaddlePaddle	ImageNet	3.0.0
RepViT	PyTorch	ImageNet	3.1.0
Res2Net50_14w_8s	PaddlePaddle	ImageNet	3.1.0
ResNeSt14	PyTorch	ImageNet	2.2.0
ResNeSt50	PyTorch	ImageNet	2.2.0
ResNeSt50	PaddlePaddle	ImageNet	3.0.0
ResNeSt101	PyTorch	ImageNet	2.2.0
ResNeSt269	PyTorch	ImageNet	2.2.0
ResNet18	PyTorch	ImageNet	2.2.0
ResNet50	PyTorch	ImageNet	2.2.0
ResNet50	PaddlePaddle	ImageNet	2.3.0
ResNet50	TensorFlow	ImageNet	3.0.0
ResNet101	PyTorch	ImageNet	2.2.0
ResNet152	PyTorch	ImageNet	2.2.0
ResNeXt50_32x4d	MindSpore	ImageNet	3.1.0
ResNeXt50_32x4d	PyTorch	ImageNet	2.2.0
ResNeXt101_32x8d	PyTorch	ImageNet	2.2.0
SE_ResNet50_vd	PaddlePaddle	ImageNet	3.1.0
SEResNeXt	PyTorch	ImageNet	2.2.0
ShuffleNetV2	PaddlePaddle	ImageNet	3.1.0
ShuffleNetV2	PyTorch	ImageNet	2.2.0
SqueezeNet	PyTorch	ImageNet	2.2.0
Swin Transformer	PaddlePaddle	ImageNet	3.0.0
Swin Transformer	PyTorch	ImageNet	2.2.0
VGG16	PaddlePaddle	ImageNet	2.3.0
VGG16	PyTorch	ImageNet	2.2.0
VGG16	TensorFlow	ImageNet	3.1.0
Wave-MLP	PyTorch	ImageNet	2.2.0
Wide_ResNet101_2	PyTorch	ImageNet	2.2.0
Xception	PaddlePaddle	ImageNet	3.1.0
Xception	PyTorch	ImageNet	2.2.0

Face Detection

Model	Framework	Dataset	IXUCA SDK
RetinaFace	PyTorch	WIDER FACE	3.0.0

Face Recognition

Model	Framework	Dataset	IXUCA SDK
ArcFace	PyTorch	CASIA-WebFaces&LFW	3.0.0
BlazeFace	PaddlePaddle	WIDER FACE	3.1.0
CosFace	PyTorch	CASIA-WebFaces&LFW	3.0.0
FaceNet	PyTorch	CASIA-WebFaces&LFW	3.0.0
FaceNet	TensorFlow	CASIA-WebFaces&LFW	3.1.0

Instance Segmentation

Model	Framework	Dataset	IXUCA SDK
SOLO	PyTorch	COCO	3.0.0
SOLOv2	PaddlePaddle	COCO	3.0.0
SOLOv2	PyTorch	COCO	3.1.0
YOLACT++	PyTorch	COCO	3.0.0

Image Generation

Model	Framework	Dataset	IXUCA SDK
DCGAN	MindSpore	ImageNet	3.0.0
Pix2Pix	PaddlePaddle	facades	3.1.0

Knowledge Distillation

Model	Framework	Dataset	IXUCA SDK
CWD	PyTorch	Cityscapes	3.0.0
RKD	PyTorch	CUB-200-2011	3.0.0
WSLD	PyTorch	ImageNet	3.1.0

Object Detection

Model	Framework	Dataset	IXUCA SDK
ATSS	PyTorch (MMDetection)	COCO	3.0.0
AutoAssign	PyTorch	COCO	2.2.0
Cascade R-CNN	PyTorch (MMDetection)	COCO	3.0.0
CenterMask2	PyTorch	COCO	4.1.1
CenterNet	PyTorch	COCO	2.2.0
CenterNet	PaddlePaddle	COCO	3.0.0
Co-DETR	PyTorch	COCO	3.1.0
CornerNet	PyTorch (MMDetection)	COCO	3.0.0
DCNV2	PyTorch (MMDetection)	COCO	3.0.0
DETR	PaddlePaddle	COCO	3.0.0
Faster R-CNN	PyTorch	COCO	2.2.0
FCOS	PaddlePaddle	COCO	3.0.0
FCOS	PyTorch	COCO	3.0.0
Mamba-YOLO	PyTorch	COCO	4.1.1
Mask R-CNN	PyTorch	COCO	2.2.0
Mask R-CNN	PaddlePaddle	COCO	2.3.0
OC_SORT	PaddlePaddle	MOT17	3.1.0
Oriented RepPoints	PyTorch	DOTA	3.1.0
PP-PicoDet	PaddlePaddle	COCO	3.1.0
PP-YOLOE	PaddlePaddle	COCO	2.3.0
PP-YOLOE+	PaddlePaddle	COCO	3.1.1
PVANet	PyTorch	COCO	2.2.0
RepPoints	PyTorch (MMDetection)	COCO	3.0.0
RetinaNet	PyTorch	COCO	2.2.0
RetinaNet	PaddlePaddle	COCO	3.0.0
RT-DETR	PyTorch	COCO	4.1.1
RTMDet	PyTorch	COCO	3.1.0
SSD	PyTorch	COCO	2.2.0
SSD	PaddlePaddle	COCO	2.3.0
SSD	TensorFlow	VOC	3.0.0
SSD	MindSpore	COCO	3.0.0
YOLOF	PyTorch	COCO	2.2.0
YOLOv3	PyTorch	COCO	2.2.0
YOLOv3	PaddlePaddle	COCO	2.3.0
YOLOv3	TensorFlow	VOC	3.0.0
YOLOv5	PaddlePaddle	COCO	3.1.1
YOLOv5	PyTorch	COCO	2.2.0
YOLOv6	PyTorch	COCO	3.0.0
YOLOv7	PyTorch	COCO	3.0.0
YOLOv8	PyTorch	COCO	3.0.0
YOLOv9	PyTorch	COCO	4.1.1
YOLOv10	PyTorch	COCO	4.1.1

3D Object Detection

Model	Framework	Dataset	IXUCA SDK
BEVFormer	PyTorch	nuScenes&CAN bus	3.0.0
CenterPoint	PyTorch	nuScenes	3.1.1
PAConv	PyTorch	S3DIS	3.1.1
Part-A2-Anchor	PyTorch	KITTI	4.1.1
Part-A2-Free	PyTorch	KITTI	4.1.1
PointNet++	PyTorch	S3DIS	3.0.0
PointPillars	PyTorch	KITTI	3.0.0
PointRCNN	PyTorch	KITTI	3.1.1
PointRCNN-IoU	PyTorch	KITTI	4.1.1
SECOND	PyTorch	KITTI	4.1.1
SECOND-IoU	PyTorch	KITTI	4.1.1

3D Reconstruction

Model	Framework	Dataset	IXUCA SDK
HashNeRF	PyTorch	fox	2.2.0

GNN (Graph Neural Network)

Model	Framework	Dataset	IXUCA SDK
GAT	PaddlePaddle	CORA	3.1.0
GCN	MindSpore	CORA & Citeseer	3.0.0
GCN	PaddlePaddle	CORA & PubMed & Citeseer	3.1.0
GraphSAGE	PaddlePaddle	Reddit	3.1.0

OCR

Model	Framework	Dataset	IXUCA SDK
CRNN	MindSpore	OCR_Recog	3.1.0
CRNN	PaddlePaddle	LMDB	2.3.0
DBNet	PyTorch	ICDAR2015	3.0.0
DBNet++	PaddlePaddle	ICDAR2015	3.1.1
DBNet++	PyTorch	ICDAR2015	3.1.0
PP-OCR-DB	PaddlePaddle	ICDAR2015	2.3.0
PP-OCR-EAST	PaddlePaddle	ICDAR2015	3.1.1
PSE	PaddlePaddle	OCR_Recog	2.3.0
SAR	PyTorch	OCR_Recog	2.2.0
SAST	PaddlePaddle	ICDAR2015	3.1.1
SATRN	PyTorch	OCR_Recog	2.2.0

Point Cloud

Model	Framework	Dataset	IXUCA SDK
Point-BERT	PyTorch	ShapeNet55 & processed ModelNet	2.2.0

Pose Estimation

Model	Framework	Dataset	IXUCA SDK
AlphaPose	PyTorch	COCO	3.0.0
HRNet	PyTorch	COCO	2.2.0
HRNet-W32	PaddlePaddle	COCO	3.1.0
OpenPose	MindSpore	COCO	3.1.0

Self-Supervised Learning

Model	Framework	Dataset	IXUCA SDK
MAE	PyTorch	ImageNet	3.0.0

Semantic Segmentation

Model	Framework	Dataset	IXUCA SDK
3D-UNet	PyTorch	kits19	2.2.0
APCNet	PyTorch	Cityscapes	2.2.0
Attention U-net	PyTorch	Cityscapes	3.0.0
BiSeNet	PyTorch	COCO	2.2.0
BiSeNetV2	PaddlePaddle	Cityscapes	3.0.0
BiSeNetV2	PyTorch	Cityscapes	3.1.1
CGNet	PyTorch	COCO	2.2.0
ContextNet	PyTorch	COCO	2.2.0
DabNet	PyTorch	COCO	2.2.0
DANet	PyTorch	COCO	2.2.0
DDRnet	PyTorch	Cityscapes	3.0.0
DeepLabV3	PyTorch	COCO	2.2.0
DeepLabV3	PaddlePaddle	Cityscapes	2.3.0
DeepLabV3	MindSpore	VOC	3.0.0
DeepLabV3+	PaddlePaddle	Cityscapes	3.0.0
DeepLabV3+	TensorFlow	Cityscapes	3.1.0
DenseASPP	PyTorch	COCO	2.2.0
DFANet	PyTorch	COCO	2.2.0
DNLNet	PaddlePaddle	Cityscapes	2.3.0
DUNet	PyTorch	COCO	2.2.0
EncNet	PyTorch	COCO	2.2.0
ENet	PyTorch	COCO	2.2.0
ERFNet	PyTorch	COCO	2.2.0
ESPNet	PyTorch	COCO	2.2.0
FastFCN	PyTorch	ADE20K	3.1.0
FastSCNN	PyTorch	COCO	2.2.0
FCN	PyTorch	COCO	2.2.0
FPENet	PyTorch	COCO	2.2.0
GCNet	PyTorch	Cityscapes	2.2.0
HardNet	PyTorch	COCO	2.2.0
ICNet	PyTorch	COCO	2.2.0
LedNet	PyTorch	COCO	2.2.0
LinkNet	PyTorch	COCO	2.2.0
Mask2Former	PyTorch	Cityscapes	3.1.0
MobileSeg	PaddlePaddle	Cityscapes	3.1.0
OCNet	PyTorch	COCO	2.2.0
OCRNet	PaddlePaddle	Cityscapes	3.1.0
OCRNet	PyTorch	Cityscapes	2.2.0
PP-HumanSegV1	PaddlePaddle	PP-HumanSeg14K	3.1.0
PP-HumanSegV2	PaddlePaddle	PP-HumanSeg14K	3.1.0
PP-LiteSeg	PaddlePaddle	Cityscapes	3.1.0
PSANet	PyTorch	COCO	2.2.0
PSPNet	PyTorch	Cityscapes	2.2.0
RefineNet	PyTorch	COCO	2.2.0
SegNet	PyTorch	COCO	2.2.0
STDC	PaddlePaddle	Cityscapes	3.1.0
STDC	PyTorch	Cityscapes	3.0.0
UNet	PyTorch	COCO	2.2.0
UNet	PaddlePaddle	Cityscapes	2.3.0
UNet++	PyTorch	DRIVE	3.0.0
VNet	TensorFlow	Hippocampus	3.0.0

Super Resolution

Model	Framework	Dataset	IXUCA SDK
basicVSR++	PyTorch	REDS	2.2.0
basicVSR	PyTorch	REDS	2.2.0
ESRGAN	PyTorch	DIV2K	2.2.0
LIIF	PyTorch	DIV2K	2.2.0
RealBasicVSR	PyTorch	REDS	2.2.0
TTSR	PyTorch	CUFED	2.2.0
TTVSR	PyTorch	REDS	2.2.0

Multi-Object Tracking

Model	Framework	Dataset	IXUCA SDK
ByteTrack	PaddlePaddle	MOT17	3.1.0
DeepSORT	PyTorch	Market-1501	3.0.0
FairMOT	PyTorch	MOT17	2.2.0

Multimodal

Model	Framework	Dataset	IXUCA SDK
BLIP	PyTorch	COCO	3.1.1
CLIP	PyTorch	CIFAR100	2.2.0
ControlNet	PyTorch	Fill50K	3.1.0
DDPM	PyTorch	CIFAR-10	3.1.0
LLaVA 1.5	PyTorch	LLaVA-Pretrain	4.1.1
L-Verse	PyTorch	ImageNet	2.2.0
Stable Diffusion 1.4	PyTorch	pokemon-images	4.1.1
Stable Diffusion 1.5	PyTorch	pokemon-images	4.1.1
Stable Diffusion 2.1	PyTorch	pokemon-images	4.1.1
Stable Diffusion 3	PyTorch	dog-example	4.1.1
Stable Diffusion XL	PyTorch	pokemon-images	4.1.1

NLP (Natural Language Processing)

Cloze Test

Model	Framework	Dataset	IXUCA SDK
GLM	PyTorch	GLMForMultiTokenCloze	2.2.0

Dialogue Generation

Model	Framework	Dataset	IXUCA SDK
CPM	PyTorch	STC	2.2.0

Language Modeling

Model	Framework	Dataset	IXUCA SDK
BART	PyTorch (Fairseq)	RTE	3.0.0
BERT NER	PyTorch	CoNLL-2003	3.0.0
BERT Pretraining	PyTorch	MLCommon Wikipedia	2.2.0
BERT Pretraining	PaddlePaddle	MNLI	2.3.0
BERT Pretraining	TensorFlow	MNLI	3.0.0
BERT Pretraining	MindSpore	SQuAD	3.0.0
BERT Text Classification	PyTorch	GLUE	3.0.0
BERT Text Summerization	PyTorch	cnn_dailymail	3.0.0
BERT Question Answering	PyTorch	SQuAD	3.0.0
GPT2-Medium-EN	PaddlePaddle	SST-2	3.1.0
RoBERTa	PyTorch (Fairseq)	RTE	3.0.0
XLNet	PaddlePaddle	SST-2	3.1.0

Text Correction

Model	Framework	Dataset	IXUCA SDK
ERNIE	PaddlePaddle	corpus	2.3.0

Translation

Model	Framework	Dataset	IXUCA SDK
Convolutional	PyTorch (Fairseq)	WMT14	3.0.0
T5	PyTorch	WMT14	2.2.0
Transformer	PaddlePaddle	WMT14	2.3.0
Transformer	PyTorch (Fairseq)	IWSLT14	3.0.0

Reinforcement Learning

Model	Framework	Dataset	IXUCA SDK
DQN	PaddlePaddle	CartPole-v0	3.1.0

Audio

Speech Recognition

Model	Framework	Dataset	IXUCA SDK
Conformer	PyTorch (WeNet)	AISHELL	2.2.0
Efficient Conformer v2	PyTorch (WeNet)	AISHELL	3.1.0
PP-ASR-Conformer	PaddlePaddle	AISHELL	3.1.0
RNN-T	PyTorch	LJSpeech	2.2.0
Transformer	PyTorch (WeNet)	AISHELL	3.0.0
U2++ Conformer	PyTorch (WeNet)	AISHELL	3.0.0
Unified Conformer	PyTorch (WeNet)	AISHELL	3.0.0

Speech Synthesis

Model	Framework	Dataset	IXUCA SDK
PP-TTS-FastSpeech2	PaddlePaddle	CSMSC	3.1.0
PP-TTS-HiFiGAN	PaddlePaddle	CSMSC	3.1.0
Tacotron2	PyTorch	LJSpeech	2.2.0
VQMIVC	PyTorch	VCTK-Corpus	2.2.0
WaveGlow	PyTorch	LJSpeech	2.2.0

Others

Graph Machine Learning

Model	Framework	Dataset	IXUCA SDK
Graph WaveNet	PyTorch	METR-LA & PEMS-BAY	2.2.0

Kolmogorov-Arnold Networks

Model	Framework	Dataset	IXUCA SDK
KAN	PyTorch	-	4.1.1

Model Pruning

Model	Framework	Dataset	IXUCA SDK
Network Slimming	PyTorch	CIFAR-10/100	3.0.0

Recommendation Systems

Model	Framework	Dataset	IXUCA SDK
DeepFM	PaddlePaddle	Criteo_Terabyte	2.3.0
DLRM	PyTorch	Criteo_Terabyte	2.2.0
DLRM	PaddlePaddle	Criteo_Terabyte	3.1.0
FFM	PaddlePaddle	Criteo_Terabyte	3.1.0
NCF	PyTorch	movielens	2.2.0
Wide&Deep	PaddlePaddle	Criteo_Terabyte	2.3.0
xDeepFM	PaddlePaddle	Criteo_Terabyte	3.1.0

容器镜像构建方式

社区用户可参考容器镜像构建说明在本地构建出能够运行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。

Name		Name	Last commit message	Last commit date
Latest commit History 931 Commits
.gitee		.gitee
audio		audio
cv		cv
docker/Iluvatar		docker/Iluvatar
docs		docs
hpc/molecular_dynamics/water_se_e2_a/tensorflow		hpc/molecular_dynamics/water_se_e2_a/tensorflow
multimodal		multimodal
nlp		nlp
others		others
reinforcement_learning/q-learning-networks/dqn/paddlepaddle		reinforcement_learning/q-learning-networks/dqn/paddlepaddle
toolbox		toolbox
.gitignore		.gitignore
CONTRIBUTING.md		CONTRIBUTING.md
LICENSE		LICENSE
README.md		README.md
RELEASE.md		RELEASE.md

License

Deep-Spark/DeepSparkHub

Folders and files

Latest commit

History

Repository files navigation