mvpzhangqiu's repositories
Yolov5_StrongSORT_OSNet
Mult-object tracking and segmentation using YOLOv5 with StrongSORT and OSNet
pytorch-CycleGAN-and-pix2pix
Image-to-Image Translation in PyTorch
nni
An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.
Onnx2TensorRT
onnx转tensorRT文件 支持动态和静态输入设置
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
TinyNeuralNetwork
TinyNeuralNetwork is an efficient and easy-to-use deep learning model compression framework.
tensorrtx
Implementation of popular deep learning networks with TensorRT network definition API
pytorch-tutorials
PyTorch tutorials.
Torch-Pruning
Structural Pruning for Model Acceleration
YOLOP
You Only Look Once for Panopitic Driving Perception.(https://arxiv.org/abs/2108.11250)
s2anet
Official code of the paper "Align Deep Features for Oriented Object Detection"
onnx-tensorrt
ONNX-TensorRT: TensorRT backend for ONNX
PyTorch-Quantization-Aware-Training
PyTorch Quantization Aware Training Example
Linux
linux
yolov5_prune
yolov5 prune,Support V2, V3, V4 and V6 versions of yolov5
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.
YOLOP-TensorRT
unofficial implementation of YOLOP TensorRT
night_light
基于opencv-python的夜间灯光识别
tensorRT
TensorRT-7 Network Lib 包括常用目标检测、关键点检测、人脸检测、OCR等 可训练自己数据
yolov5_tensorrt_int8_tools
tensorrt int8 量化yolov5 onnx模型
detectron2
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
unet-pytorch
这是一个unet-pytorch的源码,可以训练自己的模型