Stack Overflow
Stack Overflow

Questions tagged [caffe2]

Caffe2 is an open-source deep learning framework, which is developed and maintained by Facebook, and is based on the Caffe framework.

Caffe2 is a deep learning framework, which is based on the framework. It is developed and maintained by Facebook, and released under the Apache 2.0 license, on GitHub.

Caffe2 is an improved / redesigned version of Caffe. A presentation by Caffe2's lead developer Yangqing Jia, held at CVPR 2015, shows some of the consideration of the redesign.

Useful links

116 questions
11
votes
1 answer

What does mask r-cnn's AP, AP50, AP70 mean?

I'm novice on r-cnn. There are term AP, AP50, AP75 on mask r-cnn paper. 50, 75 is small postfix, but I can't make it small, sorry. Anyway the paper says it is averaged over IOU thresholds. For AP50, only candidates over 50% region comparing ground…
semenbari
  • 725
  • 1
  • 8
  • 22
11
votes
1 answer

Loading ONNX Model in Java

I have a trained PyTorch model that I would now like to export to Caffe2 using ONNX. This part seems fairly simple and well documented. However, I now want to "load" that model into a Java program in order to perform predictions within my program (a…
igodfried
  • 877
  • 9
  • 22
8
votes
1 answer

caffe2 Tensor assignment, construction or copy

This is a long shot, if you think the question is too localized, please do vote to close. I have searched on the caffe2 github repository, opened an issue asking the same question, opened another issue at the caffe2_ccp_tutorials repository because…
Ælex
  • 14,432
  • 20
  • 88
  • 129
6
votes
0 answers

AttributeError: module 'caffe2.python._import_c_extension' has no attribute 'get_cudnn_version'

I'm struggling with this error for a few days now and I have no clue how to solve this. This is the error I get when I try to import the core module from the Caffe2 package. from caffe2.python import core AttributeError: module…
247errorsnl
  • 157
  • 12
6
votes
0 answers

Install caffe2 in Google Colaboratory with GPU support

Is it possible to install caffe2 (not caffe) in Google Colaboratory with GPU support? I tried in many ways, following installation instructions from Caffe2, but I couldn't succeed to install with GPU support. When I test caffe2 installation, I get…
RomRoc
  • 1,465
  • 2
  • 12
  • 12
5
votes
2 answers

Caffe2: Load ONNX model, and inference single threaded on multi-core host / docker

I'm having trouble running inference on a model in docker when the host has several cores. The model is exported via PyTorch 1.0 ONNX exporter: torch.onnx.export(pytorch_net, dummyseq, ONNX_MODEL_PATH) Starting the model server (wrapped in Flask)…
NegatioN
  • 667
  • 2
  • 9
  • 24
5
votes
0 answers

Convert nngraph model to nn model

I want to convert the pix2pix Image to Image translation model (https://github.com/phillipi/pix2pix) which is built using nngraph. When I try to convert it to caffe model using torch to caffe tool…
Ruppesh Nalwaya
  • 1,409
  • 2
  • 14
  • 22
4
votes
1 answer

Why use Caffe2 or Core-ML instead of LibTorch(.pt file) on iOS?

It seems like there are several ways to run Pytorch models on iOS. PyTorch(.pt) -> onnx -> caffe2 PyTorch(.pt) -> onnx -> Core-ML (.mlmodel) PyTorch(.pt) -> LibTorch (.pt) PyTorch Mobile? What is the difference between the above methods? Why…
Kjyong
  • 195
  • 1
  • 8
4
votes
1 answer

Running Detectron2 inference in Caffe2

I have a Detectron2 .pth model that I converted successfully to Caffe2 .pb via the Detectron2 tools functionality located here: https://github.com/facebookresearch/detectron2/blob/master/tools/caffe2_converter.py As recommended, used the --run-eval…
gg19
  • 41
  • 1
4
votes
1 answer

Android NDK make. Hundreds of "undefined reference error"s

When trying to build this sample project https://github.com/caffe2/AICamera/tree/master/app/src/main/cpp I get hundres of error: undefined references. Here are the first few lines of output: FAILURE: Build failed with an exception. * What went…
Aidan Rosswood
  • 1,212
  • 1
  • 10
  • 21
4
votes
1 answer

Python/Caffe2: ImportError: No module named tools.setup_helpers.env

I cannot install the Python caffe2 module from the PyPi repository. When issuing sudo pip install caffe2 the following error occurs: $ sudo pip install caffe2 The directory '/home/user/.cache/pip/http' or its parent directory is not owned by the…
jfleach
  • 501
  • 1
  • 8
  • 21
4
votes
4 answers

Place some nodes of the same network on GPU and others on CPU?

When defining a network in Caffe/Caffe2, can you place some of the nodes on the CPU and others on GPU? If so, how? (If your answer pertains a specific version of Caffe, please specify which)
MWB
  • 11,740
  • 6
  • 46
  • 91
3
votes
2 answers

PyTorch to ONNX export, ATen operators not supported, onnxruntime hangs out

I want to export roberta-base based language model to ONNX format. The model uses ROBERTA embeddings and performs text classification task. from torch import nn import torch.onnx import onnx import onnxruntime import torch import transformers from…
Alexander Borochkin
  • 4,249
  • 7
  • 38
  • 53
3
votes
0 answers

Cannot open lmdb db file in Caffe2

I installed Caffe2 from source for Ubuntu. I don't use Anaconda and my python is python2. which python gives me /usr/bin/python I can import lmdb in python. I can run tutorials from this page My problem in testing one of the examples is "Can't…
batuman
  • 7,066
  • 26
  • 107
  • 229
3
votes
0 answers

LMBD error on caffe2

I am trying to learn caffe2 and I am trying to run the basic MNIST tutorial that is provided. But i am getting the following error enforce fail at db.h:190] db_. Cannot open db: of type lmdb Error from operator: I am able to download the MNIST…
user2925111
  • 31
  • 2
1
2 3 4 5 6 7 8