All Projects → inisis → Brocolli

inisis / Brocolli

pytorch 2 caffe

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Brocolli

Mobilefacenet
Stars: ✭ 114 (-24%)
Mutual labels:  caffe
Xlearning
AI on Hadoop
Stars: ✭ 1,709 (+1039.33%)
Mutual labels:  caffe
Mobilenet Ssd
Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.
Stars: ✭ 1,805 (+1103.33%)
Mutual labels:  caffe
Trojannn
Trojan Attack on Neural Network
Stars: ✭ 119 (-20.67%)
Mutual labels:  caffe
Videoauidt
📹 一个短视频APP视频内容安全审核的思路调研及实现汇总
Stars: ✭ 129 (-14%)
Mutual labels:  caffe
Flownet2 Docker
Dockerfile and runscripts for FlowNet 2.0 (estimation of optical flow)
Stars: ✭ 137 (-8.67%)
Mutual labels:  caffe
Facemaskdetection
开源人脸口罩检测模型和数据 Detect faces and determine whether people are wearing mask.
Stars: ✭ 1,677 (+1018%)
Mutual labels:  caffe
Caffe Mobilenet
A caffe implementation of mobilenet's depthwise convolution layer.
Stars: ✭ 146 (-2.67%)
Mutual labels:  caffe
Noiseface
Noise-Tolerant Paradigm for Training Face Recognition CNNs
Stars: ✭ 132 (-12%)
Mutual labels:  caffe
Anomaly Detection In Surveillance Videos
Real-World Anomaly Detection in Surveillance Videos
Stars: ✭ 139 (-7.33%)
Mutual labels:  caffe
Resnet On Cifar10
Reimplementation ResNet on cifar10 with caffe
Stars: ✭ 123 (-18%)
Mutual labels:  caffe
Darknet2caffe
Convert darknet weights to caffemodel
Stars: ✭ 127 (-15.33%)
Mutual labels:  caffe
Splitbrainauto
Split-Brain Autoencoders: Unsupervised Learning by Cross-Channel Prediction. In CVPR, 2017.
Stars: ✭ 137 (-8.67%)
Mutual labels:  caffe
Dlcookbook Dlbs
Deep Learning Benchmarking Suite
Stars: ✭ 114 (-24%)
Mutual labels:  caffe
Low Rank Bilinear Pooling
Fine-grained classification via second order statistics in a compact end-to-end trainable model
Stars: ✭ 145 (-3.33%)
Mutual labels:  caffe
Sphereface
Implementation for <SphereFace: Deep Hypersphere Embedding for Face Recognition> in CVPR'17.
Stars: ✭ 1,483 (+888.67%)
Mutual labels:  caffe
Facerecognition
This is an implematation project of face detection and recognition. The face detection using MTCNN algorithm, and recognition using LightenenCNN algorithm.
Stars: ✭ 137 (-8.67%)
Mutual labels:  caffe
Turkce Yapay Zeka Kaynaklari
Türkiye'de yapılan derin öğrenme (deep learning) ve makine öğrenmesi (machine learning) çalışmalarının derlendiği sayfa.
Stars: ✭ 1,900 (+1166.67%)
Mutual labels:  caffe
Ncnn
ncnn is a high-performance neural network inference framework optimized for the mobile platform
Stars: ✭ 13,376 (+8817.33%)
Mutual labels:  caffe
Mobilenetv2
Stars: ✭ 138 (-8%)
Mutual labels:  caffe

brocolli

a pytorch to caffe && tensorrt model converter, it learns from MMdnn && PytorchConverter. MMdnn only provides tools for pytorch to onnx conversion, but our tool provides direct conversion from pytorch to caffe && tensorrt

How to

run local

  • git clone https://github.com/inisis/caffe (only for valiation if u don't need verify your results, skip this)
  • cd caffe && (revise your dependency to caffe 3) && make pycaffe
  • pip3 install torch==0.4.0 torchvision==0.2.0
  • export PYTHONPATH=$PYTHONPATH:/your/path/to/brocolli/
  • put your pytorch model in tests/pytorch_model folder
  • python3 pytorch_model_converter.py
  • python3 ssd_layer.py example.json pytorch_model/best.pth.prototxt new.prototxt
  • python3 caffe_test.py
  • python3 pytorch_test.py

run container

  • docker pull yaphets4desmond/pytorch_converter_stable

Notice

Curently supported layers

  • Conv
  • PRelu
  • MaxPooling
  • Sigmoid
  • BatchNormalization
  • Relu
  • LeakyRelu
  • Add
  • AvgPool
  • Flatten
  • FullyConnected
  • Dropout
  • Softmax
  • Upsample
  • Permute
  • Concat

Curently supported network

Contact

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].