PlumedSerpent

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

CPlusPlusThings

C++那些事

External-Attention-pytorch

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

FasterTransformer

Transformer related optimization, including BERT, GPT

SJTUThesis

上海交通大学 LaTeX 论文模板 | Shanghai Jiao Tong University LaTeX Thesis Template

Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

Awesome-Pruning

A curated list of neural network pruning resources.

micronet

micronet, a model compression and deploy lib. compression: 1、quantization: quantization-aware-training(QAT), High-Bit(>2b)(DoReFa/Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference)、Low-Bit(≤2b)/Ternary and Binary(TWN/BNN/XNOR-Net); post-training-quantization(PTQ), 8-bit(tensorrt); 2、 pruning: normal、regular and group convolutional channel pruning; 3、 group convolution structure; 4、batch-normalization fuse for quantization. deploy: tensorrt, fp32/fp16/int8(ptq-calibration)、op-adapt(upsample)、dynamic_shape

awesome-model-quantization

A list of papers, docs, codes about model quantization. This repo is aimed to provide the info for model quantization research, we are continuously improving the project. Welcome to PR the works (papers, repositories) that are missed by the repo.

mmrazor

OpenMMLab Model Compression Toolbox and Benchmark.

MQBench

Model Quantization Benchmark

awesome-chinese-legal-resources

📝 An Awesome Collection of Chinese Legal Dataset and Relevant Resources. 致力于收集全面的中文法律数据源

wisdomInterrogatory

shrinkbench

PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.

APoT_Quantization

PyTorch implementation for the APoT quantization (ICLR 2020)

TensorRT-DETR

:zap::zap::zap:NVIDIA-阿里2021 TRT比赛 `二等奖` 代码提交 团队：美迪康 AI Lab :rocket::rocket::rocket:

UPFlow_pytorch

PyTorch implementation of UPFlow (unsupervised optical flow learning)

MAE-Lite

Official implement for ICML2023 paper: "A Closer Look at Self-Supervised Lightweight Vision Transformers"

Zero-Haruhi

The plan which extend ChatHaruhi into Zero-shot Roleplaying model

2020-CCF-Crowd-Flow-Prediction

科技战疫-大数据公益挑战赛-DataFountain重点区域人群密度预测 第1名方案

Awesome-Crowd-Localization

Awesome Crowd Localization

softsort

Code for "SoftSort: A Continuous Relaxation for the argsort Operator", ICML 2020.

Model_Compression_For_YOLOV3-V4

In this repository using the sparse training, group channel pruning and knowledge distilling for YOLOV4,

CUDA-GMM-MultiGPU

CUDA implementation of data clustering using expectation maximization with a Gaussian mixture model. Supports multiple GPUs on a single node.

ResNet38-Semantic-Segmentation

Semantic Segmentation Network based on ResNet38

LCTR

[AAAI 2022] Pytorch implementation of "LCTR: On Awakening the Local Continuity of Transformer for Weakly Supervised Object Localization".

PPE-Detection-Pose

Pose guided anchoring for detecting proper use of personal protective equipment

social-distance-detector

Winter of Code (WoC) projects and products related to Deep Fusion AI!

WAGEUBN-pytorch

ggml-cuda-experiments

CUDA Experimients