tensorrt

2熱度

2回答

將TensorRT輸出轉換爲Caffe

我的網絡包含一些不受當前張量RT支持的特定圖層。所以我想在tensorRT上運行conv層和pooling層，然後使用tensorRT的輸出作爲包含一些特定圖層的caffe模型的輸入。是否有一些我可以參考的API或示例代碼？謝謝

0熱度

2回答

將tensorflow模型轉換爲caffe模型的任何代碼？

我的最終目標是使用TensorRT優化我的部署模型。我正在做tensorflow中的實驗。但是TensorRT需要caffemodel格式的輸入模型。

0熱度

1回答

自定義圖層API（TensorRT 2.1）的簡單示例？

我正在使用TensorRT 2.1並希望實現一個簡單的自定義圖層。（我們的目標是一個嵌入系統上使用3210運行Single Shot Detector。）爲了實踐，我想使一個Inc層（只是加入1.0到輸入張量的值，並保持尺寸相同）。我執行Inc類繼class Reshape : public Iplugin後sampleFasterRNN.cpp例子。除了getOutputDimensi

0熱度

1回答

tensorrt位置「nvinfer.h」在哪裏？

我一直在嘗試編譯桌面主機上的基本tensorRT項目 - 因爲現在的源從字面上就在下面： #include <nvinfer.h> class Logger : nvinfer1::public ILogger { } glogger; 在運行make，不過，我收到以下消息： fatal error: nvinfer.h: No such file or directory #inc

6熱度

2回答

使用NVIDIA TensorRT推理引擎運行Tensorflow

我想使用NVIDIA TensorRT來運行我的Tensorflow模型。當然，TensorRT支持Caffe prototxt網絡描述符文件。我無法找到將Tensorflow模型轉換爲Caffe模型的源代碼。有沒有解決方法？