YirongMao/TensorRT-Custom-Plugin

Stars
25
Rank 952,241 (Top 19 %)
Language
C++
Created almost 4 years ago
Updated about 3 years ago

YirongMao/TensorRT-Custom-Plugin

YirongMao

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

This repository describes how to add a custom TensorRT plugin in c++ and python

softmax_variants

PyTorch code for softmax variants: center loss, cosface loss, large-margin gaussian mixture, COCOLoss, ring loss

SPDNet

This is an unofficial PyTorch implementation for paper "A Riemannian Network for SPD Matrix Learning", AAAI 2017

NAN

This is an unofficial PyTorch 1.0.1 implementation of the papr Neural Aggregation Network for Video Face Recognition. CVPR 2017

COSONet

The source code for the paper: Yirong Mao, Ruiping Wang, Shiguang Shan, Xilin Chen. COSONet: Compact Second-Order Network for Video Face Recognition. ACCV 2018