lrain-CN's starred repositories
QQMusicSpider
基于Scrapy的QQ音乐爬虫(QQ Music Spider)，爬取歌曲信息、歌词、精彩评论等，并且分享了QQ音乐中排名前6400名的内地和港台歌手的49万+的音乐语料
CPM-1-Generate
Chinese Pre-Trained Language Models (CPM-LM) Version-I
BackgroundMattingV2
Real-Time High-Resolution Background Matting
AutomaticVideoClipAndMixing
视频自动混剪模型，输入主题句，混剪出围绕主题句的一段视频，核心-语义句子匹配模型
stream-compiler
A video editor based on a human writeable script language
Video2Description
Video to Text: Natural language description generator for some given video. [Video Captioning]
auto-editor
Auto-Editor: Efficient media analysis and rendering
Library_Home_Builder
A PyClone Blender asset library designed to help with architectural and interior design
lite.ai.toolkit
🛠 A lite C++ toolkit of awesome AI models, support ONNXRuntime, MNN, TNN, NCNN and TensorRT.
RepLKNet-pytorch
Scaling Up Your Kernels to 31x31: Revisiting Large Kernel Design in CNNs (CVPR 2022)
pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
AOT-GAN-for-Inpainting
[TVCG'2023] AOT-GAN for High-Resolution Image Inpainting (codebase for image inpainting)
pytorch-summary
Model summary in PyTorch similar to `model.summary()` in Keras
pytorchviz
A small package to create visualizations of PyTorch execution graphs
DepthShrinker
[ICML 2022] "DepthShrinker: A New Compression Paradigm Towards Boosting Real-Hardware Efficiency of Compact Neural Networks", by Yonggan Fu, Haichuan Yang, Jiayi Yuan, Meng Li, Cheng Wan, Raghuraman Krishnamoorthi, Vikas Chandra, Yingyan Lin
mmdetection
OpenMMLab Detection Toolbox and Benchmark
Computer-Vision-Leaderboard
Comparison of famous convolutional neural network models
CSWin-Transformer
CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022
sota-backbones
A collection of SOTA Image Classification Models in PyTorch