shiyuan0806's repositories
STDC-Seg
Source Code of our CVPR2021 paper "Rethinking BiSeNet For Real-time Semantic Segmentation"
torch2trt
An easy to use PyTorch to TensorRT converter
ByteTrack
ByteTrack: Multi-Object Tracking by Associating Every Detection Box
PSGAN-NCNN
PSGAN running with ncnn⚡妆容迁移/仿妆⚡Imitation Makeup/Makeup Transfer⚡
yolov5-deepsort-tensorrt
A c++ implementation of yolov5 and deepsort
yolov5_prune
yolov5剪枝，支持v2,v3,v4版本的yolov5
YoLov5-TensorRT-NMS
Implement yolov5 with Tensorrt C++ api, and integrate batchedNMSPlugin. A Python wrapper is also provided.
img2pose
The official PyTorch implementation of img2pose: Face Alignment and Detection via 6DoF, Face Pose Estimation - CVPR 2021
tensorrtx
Implementation of popular deep learning networks with TensorRT network definition API
cvat
Powerful and efficient Computer Vision Annotation Tool (CVAT)
YOLOX-deepstream
deploy yolox algorithm use deepstream
face-makeup.PyTorch
Lip and hair color editor using face parsing maps.
flexible-yolov5
More readable and flexible yolov5 with more backbone(resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer) and (cbam，dcn and so on), and tensorrt
cuda_utils
CUDA utils API
CenterNet2
Two-stage CenterNet
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
PaddleSeg
End-to-end image segmentation kit based on PaddlePaddle.
Yolov5-Deepsort
最新版本yolov5+deepsort目标检测和追踪，支持5.0版本可训练自己数据集
QueryInst
QueryInst: Parallelly Supervised Mask Query for Instance Segmentation (https://arxiv.org/abs/2105.01928)
98-FaceLandmarks
98 landmark detection
Yet-Another-EfficientDet-Pytorch
The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
PSGAN
PyTorch code for "PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer" (CVPR 2020 Oral)
XNNPACK
High-efficiency floating-point neural network inference operators for mobile, server, and Web
mobilenet-ssd-snpe
mobilenet-ssd snpe demo
ScaledYOLOv4
:fire::fire::fire: Scaled-YOLOv4训练自己的数据集详细教程PDF，关于paper解读请联系小编获取PDF文档