8000 GitHub - bazige/DeepSparkHub: DeepSparkHub selects hundreds of application algorithms and models, covering various fields of AI and general-purpose computing, to support the mainstream intelligent computing scenarios.
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
/ DeepSparkHub Public
forked from Deep-Spark/DeepSparkHub

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
0 stars 12 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

bazige/DeepSparkHub

BranchesTags
 
 

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
DenseNet121 PaddlePaddle ImageNet
DenseNet201 PyTorch ImageNet
DPN92 PyTorch ImageNet
DPN107 PyTorch ImageNet
ECA_MobileNet_V2 PyTorch ImageNet
ECA_RESNET152 PyTorch ImageNet
Efficientb4 PyTorch ImageNet
EfficientNetB0 PaddlePaddle 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
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
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
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 PyTorch COCO
YOLOv6   PyTorch COCO
YOLOv7 PyTorch COCO
YOLOv8   PyTorch COCO

3D Object Detection

模型名称 框架 数据集
BEVFormer PyTorch nuScenes&CAN bus
CenterPoint PyTorch nuScenes
PointNet++   PyTorch S3DIS
PointPillars 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
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
FPE 5D32 Net 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

Multimodal

模型名称 框架 数据集
BLIP PyTorch COCO
CLIP PyTorch CIFAR100
ControlNet PyTorch Fill50K
DDPM PyTorch CIFAR-10
L-Verse PyTorch ImageNet
Stable Diffusion 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)

模型名称 框架 数据集
ChatGLM-6B PyTorch (DeepSpeed) ADGEN & chatglm-6b
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 SFT PyTorch (Megatron-DeepSpeed) gpt_small-117M

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

Reinforement 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。

交流

请联系 contact@deepspark.org.cn。

贡献

请参见 DeepSparkHub Contributing Guidelines。

许可证

本项目许可证遵循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

Stars

0 stars

Watchers

0 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages

  • Python 87.4%
  • C++ 5.9%
  • Cuda 3.5%
  • Shell 1.8%
  • C 1.2%
  • Perl 0.1%
  • Other 0.1%
0