xiezheng-cs

MobileNet-SSD

Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.

PytorchToCaffe

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs

pytorch2ncnn

pytorch_converter

pytorch2ncnn

convert pytorch model to ncnn

PytorchConverter

Pytorch model to caffe & ncnn

draw-roc-curves

draw roc curves according to the resulted files

DrawROC

Simple Matlab Script to draw a ROC Curve from array of scores and labels

AutoMetric

draw ROC,PR curve and calculate AUC MAP IoU for image semantic segmentation problem

pytorch2caffe

Convert PyTorch model to Caffemodel

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

pytorch-distributed-example

haq

[CVPR 2019, Oral] HAQ: Hardware-Aware Automated Quantization with Mixed Precision

pytorch-OpCounter

Count the FLOPs of your PyTorch model.

pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

mobilenetv3

mobilenetv3 with pytorch，provide pre-train model

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

EfficientNet-PyTorch

A PyTorch implementation of EfficientNet

mathAI

一个拍照做题程序。输入一张包含数学计算题的图片，输出识别出的数学计算式以及计算结果。This is a mathematic expression recognition project.

pywick

High-level batteries-included neural network training library for Pytorch

Fabrik

:factory: Collaboratively build, visualize, and design neural nets in browser

MTCNN-Tensorflow

Reproduce MTCNN using Tensorflow

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

models

Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.

MTCNN

Realtime Face Detection and Head pose estimation on Windows、Ubuntu、Mac、Android and iOS

tvm_mobilefacenet

PySnooper

Never use print for debugging again

SKNet_pytorch

A reimplementation of the SKNet(Selective Kernel Networks) by pytorch

SKNet

Code for our CVPR 2019 paper: Selective Kernel Networks; See zhihu：https://zhuanlan.zhihu.com/p/59690223

OctaveConv

A MXNet Implementation for Drop an Octave

waifu2x

Image Super-Resolution for Anime-Style Art