Pinned Repositories
ATen
ATen: A TENsor library for C++11
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
dgd_person_reid
Domain Guided Dropout for Person Re-identification
Efficient-PyTorch
My best practice of training large dataset using PyTorch.
ESGR
The source code for BMVC 2018 paper - Exemplar-Supported Generative Reproduction for Class Incremental Learning
NAN
This is an unofficial PyTorch 1.0.1 implementation of the papr Neural Aggregation Network for Video Face Recognition. CVPR 2017
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
TensorRT-Custom-Plugin
This repository describes how to add a custom TensorRT plugin in c++ and python
torch2trt
Add new plugin layer for torch2trt package
YirongMao's Repositories
YirongMao/softmax_variants
PyTorch code for softmax variants: center loss, cosface loss, large-margin gaussian mixture, COCOLoss, ring loss
YirongMao/SPDNet
This is an unofficial PyTorch implementation for paper "A Riemannian Network for SPD Matrix Learning", AAAI 2017
YirongMao/TensorRT-Custom-Plugin
This repository describes how to add a custom TensorRT plugin in c++ and python
YirongMao/NAN
This is an unofficial PyTorch 1.0.1 implementation of the papr Neural Aggregation Network for Video Face Recognition. CVPR 2017
YirongMao/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
YirongMao/ATen
ATen: A TENsor library for C++11
YirongMao/dgd_person_reid
Domain Guided Dropout for Person Re-identification
YirongMao/Efficient-PyTorch
My best practice of training large dataset using PyTorch.
YirongMao/ESGR
The source code for BMVC 2018 paper - Exemplar-Supported Generative Reproduction for Class Incremental Learning
YirongMao/torch2trt
Add new plugin layer for torch2trt package
YirongMao/Probabilistic-Graphical-Models
Code for Implementation, Inference, and Learning of Bayesian and Markov Networks along with some practical examples.
YirongMao/TensorRT
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.