edpzou's repositories
attention-module
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
awesome-model-quantization
A list of papers, docs, codes about model quantization. This repo is aimed to provide the info for model quantization research, we are continuously improving the project. Welcome to PR the works (papers, repositories) that are missed by the repo.
Awesome-Pruning
A curated list of neural network pruning resources.
byterun
A Python implementation of a Python bytecode runner
C3D
C3D is a modified version of BVLC caffe to support 3D ConvNets.
cytoscape.js-dagre
The Dagre layout for DAGs and trees for Cytoscape.js
dagre-d3
A D3-based renderer for Dagre
DouZero
[ICML 2021] DouZero: Mastering DouDizhu with Self-Play Deep Reinforcement Learning | 斗地主AI
export_llama_to_onnx
export llama to onnx
gpu-monitor
Script to remotely check GPU servers for free GPUs
mobula
A Lightweight & Flexible Deep Learning (Neural Network) Framework in Python
MQBench
Model Quantization Benchmark
netron
Visualizer for deep learning and machine learning models
netscope
Neural network visualizer
ohmyzsh
🙃 A delightful community-driven (with 1800+ contributors) framework for managing your zsh configuration. Includes 300+ optional plugins (rails, git, OSX, hub, docker, homebrew, node, php, python, etc), 140+ themes to spice up your morning, and an auto-update tool so that makes it easy to keep up with the latest updates from the community.
onnx
Open Neural Network Exchange
onnx-docker
Dockerfiles and scripts for ONNX container images
protobuf
Protocol Buffers - Google's data interchange format
pymlir
Python interface for MLIR - the Multi-Level Intermediate Representation
SKNet
Code for our CVPR 2019 paper: Selective Kernel Networks; See zhihu：https://zhuanlan.zhihu.com/p/59690223
sphereface
Implementation for <SphereFace: Deep Hypersphere Embedding for Face Recognition> in CVPR'17.
ssd_torch
fork from https://github.com/amdegroot/ssd.pytorch
tensorflow-onnx
Convert TensorFlow models to ONNX
TensorRT
TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.
tfrecord
TFRecord reader for PyTorch
tvm
Open deep learning compiler stack for cpu, gpu and specialized accelerators
vision
Datasets, Transforms and Models specific to Computer Vision
yolact
A simple, fully convolutional model for real-time instance segmentation.