Giter Site home page Giter Site logo

xiaozhiob / deepsparkhub Goto Github PK

View Code? Open in Web Editor NEW

This project forked from deep-spark/deepsparkhub

0.0 0.0 0.0 237.44 MB

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 License 2.0

Shell 1.66% JavaScript 0.01% C++ 6.38% Python 87.04% Perl 0.07% C 1.03% MATLAB 0.03% CSS 0.01% Objective-C++ 0.01% Cuda 3.56% Makefile 0.01% HTML 0.01% CMake 0.04% Batchfile 0.02% Jupyter Notebook 0.06% GDB 0.01% Cython 0.01% Dockerfile 0.05% Java 0.01%

deepsparkhub's Introduction

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

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

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)

模型名称 框架 数据集
Baichuan2-7B PyTorch baichuan2-7b-base
Bloom-7B1 PyTorch (Firefly) school_math_0.25M & bloom-7b1
ChatGLM-6B PyTorch (DeepSpeed) ADGEN & chatglm-6b
ChatGLM2-6B SFT PyTorch ADGEN & chatglm2-6b
LLaMA-7B PyTorch (Colossal-AI) llama-7b-hf
Llama-2-7B PyTorch (Megatron-DeepSpeed) Bookcorpus
Llama-2-7B Reward Model Finetuning PyTorch (DeepSpeed) Dahoas/rm-static
Llama-7B RLHF PyTorch (Megatron-DeepSpeed) llama2-7b&tiny-llama
Llama-2-7B SFT PyTorch (Megatron-DeepSpeed) gpt_small-117M
QWen-7B PyTorch (Firefly) qwen-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。

许可证

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

deepsparkhub's People

Contributors

rexlee8776 avatar miracleli1997 avatar tciluvatar avatar wacyfdyy avatar jeffguo2022 avatar ruili34 avatar haoyanlong avatar li-shi-kun avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.