inisis / Brocolli
pytorch 2 caffe
Stars: ✭ 150
Programming Languages
python
139335 projects - #7 most used programming language
Projects that are alternatives of or similar to Brocolli
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
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
Splitbrainauto
Split-Brain Autoencoders: Unsupervised Learning by Cross-Channel Prediction. In CVPR, 2017.
Stars: ✭ 137 (-8.67%)
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
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
- SSD Pretrained Weights 提取码:kiaf
- ResNet
- MTCNN Pretrained Weights
- YOLOV3
- retinanet
Contact
- Desmond [email protected]
- qq group: 597059928
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].