{"id":19001218,"url":"https://github.com/oneflow-inc/models","last_synced_at":"2025-04-04T22:05:30.793Z","repository":{"id":37826545,"uuid":"321305310","full_name":"Oneflow-Inc/models","owner":"Oneflow-Inc","description":"Models and examples built with OneFlow","archived":false,"fork":false,"pushed_at":"2024-10-16T09:41:54.000Z","size":161297,"stargazers_count":97,"open_issues_count":50,"forks_count":38,"subscribers_count":40,"default_branch":"main","last_synced_at":"2025-03-28T21:05:22.006Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":null,"language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/Oneflow-Inc.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2020-12-14T09:59:52.000Z","updated_at":"2025-03-25T05:41:13.000Z","dependencies_parsed_at":"2024-06-02T14:01:01.952Z","dependency_job_id":"2f7d29c5-2d0a-4caf-8a08-6ca8860df2f6","html_url":"https://github.com/Oneflow-Inc/models","commit_stats":null,"previous_names":[],"tags_count":1,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/Oneflow-Inc%2Fmodels","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/Oneflow-Inc%2Fmodels/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/Oneflow-Inc%2Fmodels/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/Oneflow-Inc%2Fmodels/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/Oneflow-Inc","download_url":"https://codeload.github.com/Oneflow-Inc/models/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":247256110,"owners_count":20909240,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2024-11-08T18:10:23.032Z","updated_at":"2025-04-04T22:05:30.776Z","avatar_url":"https://github.com/Oneflow-Inc.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"# OneFlow-Models\n**Models and examples implement with OneFlow(version \u003e= 0.5.0).**\n\n## Introduction\n**English** | [简体中文](/README_zh-CN.md)\n\nOneFlow-Models is an open source repo which contains official implementation of different models built on OneFlow. In each model, we provide at least two scripts `train.sh` and `infer.sh` for a quick start. For each model, we provide a detailed `README` to introduce the usage of this model.\n\n## Features\n- various models and pretrained weight\n- easy use for beginners\n\n## Quick Start\nPlease check our the following **demos** for a quick start\n- **image classification** [quick start lenet demo](Demo/quick_start_demo_lenet/lenet.py)\n- **speaker recognition** [speaker identification demo](Demo/speaker_identification_demo)\n\n## Model List\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Image Classification \u003c/b\u003e \u003c/summary\u003e\n\n  - [Lenet](https://github.com/Oneflow-Inc/models/blob/main/Demo/quick_start_demo_lenet/lenet.py)\n  - [Alexnet](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/alexnet)\n  - [VGG16/19](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/vgg)\n  - [Resnet50](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/resnet50)\n  - [InceptionV3](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/inception_v3)\n  - [Densenet](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/densenet)\n  - [Resnext50_32x4d](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/resnext50_32x4d)\n  - [Shufflenetv2](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/shufflenetv2)\n  - [MobilenetV2](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/mobilenetv2)\n  - [mobilenetv3](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/mobilenetv3)\n  - [Ghostnet](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/ghostnet)\n  - [RepVGG](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/repvgg)\n  - [DLA](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/DLA)\n  - [PoseNet](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/poseNet)\n  - [Scnet](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/scnet)\n  - [Mnasnet](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/mnasnet)\n  - [ViT](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/image/ViT)\n\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Video Classification \u003c/b\u003e \u003c/summary\u003e\n\n- [TSN](https://github.com/Oneflow-Inc/models/tree/main/Vision/classification/video/TSN)\n\n\u003c/details\u003e\n\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Object Detection \u003c/b\u003e \u003c/summary\u003e\n  \n- [CSRNet](https://github.com/Oneflow-Inc/models/tree/main/Vision/detection/CSRNet)\n\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Semantic Segmentation \u003c/b\u003e \u003c/summary\u003e\n\n- [FODDet](https://github.com/Oneflow-Inc/models/tree/main/Vision/segmentation/FODDet)\n- [FaceSeg](https://github.com/Oneflow-Inc/models/tree/main/Vision/segmentation/FaceSeg)\n- [U-Net](https://github.com/Oneflow-Inc/models/tree/main/Vision/segmentation/U-Net)\n\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Generative Adversarial Networks \u003c/b\u003e \u003c/summary\u003e\n\n- [DCGAN](https://github.com/Oneflow-Inc/models/tree/main/Vision/gan/DCGAN)\n- [SRGAN](https://github.com/Oneflow-Inc/models/tree/main/Vision/gan/SRGAN)\n- [Pix2Pix](https://github.com/Oneflow-Inc/models/tree/main/Vision/gan/Pix2Pix)\n- [CycleGAN](https://github.com/Oneflow-Inc/models/tree/main/Vision/gan/CycleGAN)\n\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Neural Style Transform \u003c/b\u003e \u003c/summary\u003e\n\n- [FastNeuralStyle](https://github.com/Oneflow-Inc/models/tree/main/Vision/style_transform/fast_neural_style)\n\n\u003c/details\u003e\n\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Person Re-identification \u003c/b\u003e \u003c/summary\u003e\n\n- [BoT](https://github.com/Oneflow-Inc/models/tree/main/Vision/reid/BoT)\n\n\u003c/details\u003e\n\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Natural Language Processing \u003c/b\u003e \u003c/summary\u003e\n\n- [RNN](https://github.com/Oneflow-Inc/models/tree/main/NLP/rnn)\n- [Seq2Seq](https://github.com/Oneflow-Inc/models/tree/main/NLP/seq2seq)\n- [LSTMText](https://github.com/Oneflow-Inc/models/tree/main/NLP/LSTMText)\n- [TextCNN](https://github.com/Oneflow-Inc/models/tree/main/NLP/TextCNN)\n- [Transformer](https://github.com/Oneflow-Inc/models/tree/main/NLP/Transformer)\n- [Bert](https://github.com/Oneflow-Inc/models/tree/main/NLP/bert-oneflow)\n- [CPT](https://github.com/Oneflow-Inc/models/tree/main/NLP/CPT)\n- [Roberta](https://github.com/Oneflow-Inc/models/tree/main/NLP/roberta)\n- [MoE](https://github.com/Oneflow-Inc/models/tree/main/NLP/MoE)\n\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Audio \u003c/b\u003e \u003c/summary\u003e\n\n- [SincNet](https://github.com/Oneflow-Inc/models/tree/main/Audio/SincNet)\n- [Wav2Letter](https://github.com/Oneflow-Inc/models/tree/main/Audio/Wav2Letter)\n- [AM_MobileNet1D](https://github.com/Oneflow-Inc/models/tree/main/Audio/AM-MobileNet1D)\n- [Speech-Emotion-Analyer](https://github.com/Oneflow-Inc/models/tree/main/Audio/Speech-Emotion-Analyzer)\n- [Speech-Transformer](https://github.com/Oneflow-Inc/models/tree/main/Audio/Speech-Transformer)\n- [CycleGAN-VC2](https://github.com/Oneflow-Inc/models/tree/main/Audio/CycleGAN-VC2)\n- [MaskCycleGAN-VC](https://github.com/Oneflow-Inc/models/tree/main/Audio/MaskCycleGAN-VC)\n- [StarGAN-VC](https://github.com/Oneflow-Inc/models/tree/main/Audio/StarGAN-VC)\n- [Adaptive_Voice_Conversion](https://github.com/Oneflow-Inc/models/tree/main/Audio/Adaptive_Voice_Conversion)\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Deep Reinforcement Learning \u003c/b\u003e \u003c/summary\u003e\n\n- [FlappyBird](https://github.com/Oneflow-Inc/models/tree/main/DeepReinforcementLearning/FlappyBird)\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Quantization Aware Training \u003c/b\u003e \u003c/summary\u003e\n\n- [Quantization](https://github.com/Oneflow-Inc/models/tree/main/Quantization)\n\u003c/details\u003e\n\n\u003cdetails\u003e\n\u003csummary\u003e \u003cb\u003e Knowledge Distillation \u003c/b\u003e \u003c/summary\u003e\n\n- [KnowledgeDistillation](https://github.com/Oneflow-Inc/models/tree/main/KnowledgeDistillation/KnowledgeDistillation)\n\u003c/details\u003e\n\n## Installation and Environment setup\n**Install Oneflow**\n\nhttps://github.com/Oneflow-Inc/oneflow#install-with-pip-package\n\n**Build custom ops from source**\n\nIn the root directory, run:\n```bash\nmkdir build\ncd build\ncmake ..\nmake -j$(nrpoc)\n```\nExample of using ops:\n```bash\nfrom ops import RoIAlign\npooler = RoIAlign(output_size=(14, 14), spatial_scale=2.0, sampling_ratio=2)\n```\n\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Foneflow-inc%2Fmodels","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Foneflow-inc%2Fmodels","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Foneflow-inc%2Fmodels/lists"}