mvpzhangqiu

yolov5prune

alfred

alfred-py: A deep learning utility library for **human**, more detail about the usage of lib to: https://zhuanlan.zhihu.com/p/341446046

computer_vision

detectron2

Detectron2 is FAIR's next-generation platform for object detection and segmentation.

leetcode

night_light

基于opencv-python的夜间灯光识别

SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

SOLOV2

unet-pytorch

这是一个unet-pytorch的源码，可以训练自己的模型

vision

Datasets, Transforms and Models specific to Computer Vision

yolov5-deepsort

基于yolov5与deepsort的目标跟踪、测速模型

YOLOv5-Multibackbone-Compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming) and Quantization (MQBench) Compression Tool Box.

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Linux

linux

nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

onnx-tensorrt

ONNX-TensorRT: TensorRT backend for ONNX

Onnx2TensorRT

onnx转tensorRT文件 支持动态和静态输入设置

pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch

PyTorch-Quantization-Aware-Training

PyTorch Quantization Aware Training Example

pytorch-tutorials

PyTorch tutorials.

s2anet

Official code of the paper "Align Deep Features for Oriented Object Detection"

tensorRT

TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据

tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API

TinyNeuralNetwork

TinyNeuralNetwork is an efficient and easy-to-use deep learning model compression framework.

Torch-Pruning

Structural Pruning for Model Acceleration

YOLOP

You Only Look Once for Panopitic Driving Perception.（https://arxiv.org/abs/2108.11250）

YOLOP-TensorRT

unofficial implementation of YOLOP TensorRT

yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5

Yolov5_StrongSORT_OSNet

Mult-object tracking and segmentation using YOLOv5 with StrongSORT and OSNet

yolov5_tensorrt_int8_tools

tensorrt int8 量化yolov5 onnx模型