Mars's repositories
Detectron.pytorch
A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
cnnimageretrieval-pytorch
CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch
Dataset-Converters
A conversion toolset between different object detection and instance segmentation annotation formats
mediapipe
Cross-platform, customizable ML solutions for live and streaming media.
models
Models and examples built with TensorFlow
pytorch-pruning
PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference and pruning for YoLo-v2
TensorRT
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.
trivialaugment
This is the official implementation of TrivialAugment and a mini-library for the application of multiple image augmentation strategies including RandAugment and TrivialAugment.
tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
