关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>
Happiness22
/
DeepSparkHub
forked from DeepSpark/DeepSparkHub
Not watched
1
0
Fork 0
Code
Releases 0 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
534 Commits
7 Branches
1.1 GiB
0 Download
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
majorli6 7484616022 !296 add reference in Nerf README.md and format 4 months ago
.gitee update .gitee/PULL_REQUEST_TEMPLATE.zh-CN.md. 9 months ago
3d-reconstruction/ngp-nerf/pytorch !296 add reference in Nerf README.md and format 4 months ago
Reinforement_Learning/DQN/paddlepaddle format Reinforement_Learning/DQN/paddlepaddle/README.md. 6 months ago
cv update SAR model datasets link 4 months ago
docker/Iluvatar !175 add deepsparkhub docker image build guide 10 months ago
docs add ImageNet dataset tips for PaddleClas training 6 months ago
gnn move graphsage and gat model location 4 months ago
hpc/molecular_dynamics/water_se_e2_a/tensorflow !239 Rename file path according to the model category 6 months ago
multimodal move controlnet location 4 months ago
nlp update requirements version 4 months ago
recommendation !286 add FFM PaddlePaddle model 5 months ago
speech !283 Improve: remove unnecessary .* hidden files 5 months ago
toolbox add schedule_1x.py 4 months ago
.gitignore !175 add deepsparkhub docker image build guide 10 months ago
CONTRIBUTING.md gitee and github docs keep same 1 year ago
LICENSE Initial Commit 1 year ago
README.md !296 add reference in Nerf README.md and format 4 months ago
README_TEMPLATE.md Initial Commit 1 year ago
RELEASE.md update RELEASE.md. 4 months ago

README.md

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
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
PointNet++   PyTorch S3DIS
PointPillars PyTorch KITTI

OCR

模型名称 框架 数据集
CRNN MindSpore OCR_Recog
CRNN PaddlePaddle LMDB
DBNet PyTorch ICDAR2015
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
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
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

模型名称 框架 数据集
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

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。

Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领域，支持主流市场智能计算落地应用场景。

Python Unity3D Asset Text Jupyter Notebook C++ other

Please read the following content carefully:

Dear OpenI User

Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.

For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》

Agree and continue
Disagree, exit