Guocheng's repositories
MNISTDiffusion
Implement a MNIST(also minimal) version of denoising diffusion probabilistic model from scratch.The model only has 4.55MB.
SoftmaxFocalLoss
An intuitive pytorch implementation of softmax focal loss which can be used to train classification and segmentation tasks.
yolov5_tinytrt
Deploy yolov5 on TensorRT with Libtinytrt ⚡️.Both x86 and ARM(NVIDIA Jetson).
yolov5_pruning
Network pruning on yolov5.
kitti_baseline
The "Hello World" approch of KITTI Benchmark
carla_leaderboard
carla leaderboard 2.0
code-for-blog
Code samples from my blog
fopnp
Foundations of Python Network Programming (Apress) — scripts and examples
fucking-algorithm
刷算法全靠套路，认准 labuladong 就够了！English version supported! Crack LeetCode, not only how, but also why.
inference_sample_collection
Sample code collection of mainstream inference frameworks in C++.
LAV
(CVPR 2022) A minimalistic mapless end-to-end stack for joint perception, prediction, planning and control for self driving.
mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
onnx2caffe
onnx模型转caffe模型的工具
open-metric-learning
Library for metric learning pipelines
opencv
Open Source Computer Vision Library
ORB_SLAM3
ORB-SLAM3: An Accurate Open-Source Library for Visual, Visual-Inertial and Multi-Map SLAM
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
pytorch-metric-learning
The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
sort
Simple, online, and realtime tracking of multiple objects in a video sequence.
tiny-tensorrt
A simple, efficient, easy-to-use Nvidia TensorRT wrapper, support c++ and python.
tinygrad
You like pytorch? You like micrograd? You love tinygrad! ❤️
torchdistill
A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆20 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.
torchmetrics
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch