tanghaifeng's repositories
centernet-opencv-onnxrun-cpp-py
分别使用OpenCV、ONNXRuntime部署CenterNet目标检测，包含C++和Python两个版本的程序
annotated_deep_learning_paper_implementations
🧑🏫 59 Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠
CCTSDB
CSUST Chinese Traffic Sign Detection Benchmark
Co-DETR
[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
DenoisingDiffusionProbabilityModel-ddpm-
This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.
EasyAnimate
📺 Generate your animation easily
EndoSLAM
EndoSLAM Dataset and an Unsupervised Monocular Visual Odometry and Depth Estimation Approach for Endoscopic Videos: Endo-SfMLearner
gt4sd-core
GT4SD, an open-source library to accelerate hypothesis generation in the scientific discovery process.
latent-diffusion
High-Resolution Image Synthesis with Latent Diffusion Models
LW-DETR
This repository is an official implementation of the paper "LW-DETR: A Transformer Replacement to YOLO for Real-Time Detection".
mae
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
MedicalGPT
MedicalGPT: Training Your Own Medical GPT Model with ChatGPT Training Pipeline. 训练医疗大模型，实现了包括增量预训练、有监督微调、RLHF(奖励建模、强化学习训练)和DPO(直接偏好优化)。
mmdetection
OpenMMLab Detection Toolbox and Benchmark
mmdetection3d
OpenMMLab's next-generation platform for general 3D object detection.
Paddle3D
A 3D computer vision development toolkit based on PaddlePaddle. It supports point-cloud object detection, segmentation, and monocular 3D object detection models.
pcl-learning
🔥PCL（Point Cloud Library）点云库学习记录
Pytorch-UNet
PyTorch implementation of the U-Net for image semantic segmentation with high quality images
TensorRT-Alpha
🔥🔥🔥TensorRT-Alpha supports YOLOv8、YOLOv7、YOLOv6、YOLOv5、YOLOv4、v3、YOLOX、YOLOR...🚀🚀🚀CUDA IS ALL YOU NEED.🍎🍎🍎It also supports end2end CUDA C acceleration and multi-batch inference.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
unet-pytorch
这是一个unet-pytorch的源码，可以训练自己的模型
unilm
Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
YOLOv6
YOLOv6: a single-stage object detection framework dedicated to industrial applications.
yolov7
Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
yolov8-opencv-onnxruntime-cpp
detection and instance segmentation of yolov8,use onnxruntime and opencv
yolov8n_onnx_tensorRT_rknn_horizon
yolov8n 部署版本，后处理用python语言和C++语言形式进行改写，便于移植不同平台（onnx、tensorRT、RKNN、Horzion）