18 Star 74 Fork 49

DeepSpark / DeepSparkHub

Create your Gitee Account
Explore and code with more than 12 million developers,Free private repositories !:)
Sign up
Clone or Download
contribute
Sync branch
Cancel
Notice: Creating folder will generate an empty file .keep, because not support in Git
Loading...
README
Apache-2.0

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)

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

交流

请联系 contact@deepspark.org.cn

贡献

请参见 DeepSparkHub Contributing Guidelines

许可证

本项目许可证遵循Apache-2.0

Apache License Version 2.0, January 2004 http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. "License" shall mean the terms and conditions for use, reproduction, and distribution as defined by Sections 1 through 9 of this document. "Licensor" shall mean the copyright owner or entity authorized by the copyright owner that is granting the License. "Legal Entity" shall mean the union of the acting entity and all other entities that control, are controlled by, or are under common control with that entity. For the purposes of this definition, "control" means (i) the power, direct or indirect, to cause the direction or management of such entity, whether by contract or otherwise, or (ii) ownership of fifty percent (50%) or more of the outstanding shares, or (iii) beneficial ownership of such entity. "You" (or "Your") shall mean an individual or Legal Entity exercising permissions granted by this License. "Source" form shall mean the preferred form for making modifications, including but not limited to software source code, documentation source, and configuration files. "Object" form shall mean any form resulting from mechanical transformation or translation of a Source form, including but not limited to compiled object code, generated documentation, and conversions to other media types. "Work" shall mean the work of authorship, whether in Source or Object form, made available under the License, as indicated by a copyright notice that is included in or attached to the work (an example is provided in the Appendix below). "Derivative Works" shall mean any work, whether in Source or Object form, that is based on (or derived from) the Work and for which the editorial revisions, annotations, elaborations, or other modifications represent, as a whole, an original work of authorship. For the purposes of this License, Derivative Works shall not include works that remain separable from, or merely link (or bind by name) to the interfaces of, the Work and Derivative Works thereof. "Contribution" shall mean any work of authorship, including the original version of the Work and any modifications or additions to that Work or Derivative Works thereof, that is intentionally submitted to Licensor for inclusion in the Work by the copyright owner or by an individual or Legal Entity authorized to submit on behalf of the copyright owner. For the purposes of this definition, "submitted" means any form of electronic, verbal, or written communication sent to the Licensor or its representatives, including but not limited to communication on electronic mailing lists, source code control systems, and issue tracking systems that are managed by, or on behalf of, the Licensor for the purpose of discussing and improving the Work, but excluding communication that is conspicuously marked or otherwise designated in writing by the copyright owner as "Not a Contribution." "Contributor" shall mean Licensor and any individual or Legal Entity on behalf of whom a Contribution has been received by Licensor and subsequently incorporated within the Work. 2. Grant of Copyright License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable copyright license to reproduce, prepare Derivative Works of, publicly display, publicly perform, sublicense, and distribute the Work and such Derivative Works in Source or Object form. 3. Grant of Patent License. Subject to the terms and conditions of this License, each Contributor hereby grants to You a perpetual, worldwide, non-exclusive, no-charge, royalty-free, irrevocable (except as stated in this section) patent license to make, have made, use, offer to sell, sell, import, and otherwise transfer the Work, where such license applies only to those patent claims licensable by such Contributor that are necessarily infringed by their Contribution(s) alone or by combination of their Contribution(s) with the Work to which such Contribution(s) was submitted. If You institute patent litigation against any entity (including a cross-claim or counterclaim in a lawsuit) alleging that the Work or a Contribution incorporated within the Work constitutes direct or contributory patent infringement, then any patent licenses granted to You under this License for that Work shall terminate as of the date such litigation is filed. 4. Redistribution. You may reproduce and distribute copies of the Work or Derivative Works thereof in any medium, with or without modifications, and in Source or Object form, provided that You meet the following conditions: (a) You must give any other recipients of the Work or Derivative Works a copy of this License; and (b) You must cause any modified files to carry prominent notices stating that You changed the files; and (c) You must retain, in the Source form of any Derivative Works that You distribute, all copyright, patent, trademark, and attribution notices from the Source form of the Work, excluding those notices that do not pertain to any part of the Derivative Works; and (d) If the Work includes a "NOTICE" text file as part of its distribution, then any Derivative Works that You distribute must include a readable copy of the attribution notices contained within such NOTICE file, excluding those notices that do not pertain to any part of the Derivative Works, in at least one of the following places: within a NOTICE text file distributed as part of the Derivative Works; within the Source form or documentation, if provided along with the Derivative Works; or, within a display generated by the Derivative Works, if and wherever such third-party notices normally appear. The contents of the NOTICE file are for informational purposes only and do not modify the License. You may add Your own attribution notices within Derivative Works that You distribute, alongside or as an addendum to the NOTICE text from the Work, provided that such additional attribution notices cannot be construed as modifying the License. You may add Your own copyright statement to Your modifications and may provide additional or different license terms and conditions for use, reproduction, or distribution of Your modifications, or for any such Derivative Works as a whole, provided Your use, reproduction, and distribution of the Work otherwise complies with the conditions stated in this License. 5. Submission of Contributions. Unless You explicitly state otherwise, any Contribution intentionally submitted for inclusion in the Work by You to the Licensor shall be under the terms and conditions of this License, without any additional terms or conditions. Notwithstanding the above, nothing herein shall supersede or modify the terms of any separate license agreement you may have executed with Licensor regarding such Contributions. 6. Trademarks. This License does not grant permission to use the trade names, trademarks, service marks, or product names of the Licensor, except as required for reasonable and customary use in describing the origin of the Work and reproducing the content of the NOTICE file. 7. Disclaimer of Warranty. Unless required by applicable law or agreed to in writing, Licensor provides the Work (and each Contributor provides its Contributions) on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, including, without limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR PURPOSE. You are solely responsible for determining the appropriateness of using or redistributing the Work and assume any risks associated with Your exercise of permissions under this License. 8. Limitation of Liability. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise, unless required by applicable law (such as deliberate and grossly negligent acts) or agreed to in writing, shall any Contributor be liable to You for damages, including any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this License or out of the use or inability to use the Work (including but not limited to damages for loss of goodwill, work stoppage, computer failure or malfunction, or any and all other commercial damages or losses), even if such Contributor has been advised of the possibility of such damages. 9. Accepting Warranty or Additional Liability. While redistributing the Work or Derivative Works thereof, You may choose to offer, and charge a fee for, acceptance of support, warranty, indemnity, or other liability obligations and/or rights consistent with this License. However, in accepting such obligations, You may act only on Your own behalf and on Your sole responsibility, not on behalf of any other Contributor, and only if You agree to indemnify, defend, and hold each Contributor harmless for any liability incurred by, or claims asserted against, such Contributor by reason of your accepting any such warranty or additional liability. END OF TERMS AND CONDITIONS APPENDIX: How to apply the Apache License to your work. To apply the Apache License to your work, attach the following boilerplate notice, with the fields enclosed by brackets "[]" replaced with your own identifying information. (Don't include the brackets!) The text should be enclosed in the appropriate comment syntax for the file format. We also recommend that a file or class name and description of purpose be included on the same "printed page" as the copyright notice for easier identification within third-party archives. Copyright [yyyy] [name of copyright owner] Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.

About

DeepSparkHub甄选上百个应用算法和模型,覆盖AI和通用计算各领域,支持主流市场智能计算落地应用场景。 expand collapse
Python and 6 more languages
Apache-2.0
Cancel

Contributors

All

Activities

Load More
can not load any more
Python
1
https://gitee.com/deep-spark/deepsparkhub.git
git@gitee.com:deep-spark/deepsparkhub.git
deep-spark
deepsparkhub
DeepSparkHub
master

Search