lewes6369 / Tensorrtwrapper
Licence: mit
TensorRT Net Wrapper
Stars: β 81
Projects that are alternatives of or similar to Tensorrtwrapper
Mobilnet ssd opencv
MobilNet-SSD object detection in opencv 3.4.1
Stars: β 64 (-20.99%)
Mutual labels: caffe
Swifttwitch
πΎ The New Twitch API for iOS; wrapped in Swift goodness πΎ
Stars: β 72 (-11.11%)
Mutual labels: wrapper
Caffe Tools
Some tools and examples for pyCaffe including LMDB I/O, custom Python layers and monitoring training error and loss.
Stars: β 78 (-3.7%)
Mutual labels: caffe
Caffe2pytorch Tsn
Transform the caffe model to pytorch model for Temporal Segment Network
Stars: β 69 (-14.81%)
Mutual labels: caffe
Phenomenon Px
β‘οΈ The fastest way to create pixel shaders.
Stars: β 77 (-4.94%)
Mutual labels: wrapper
Ssd Models
ζζιζ£ζ΅ε¨ηι¨ζ§η»ζζδΈζ₯make lightweight caffe-ssd great again
Stars: β 62 (-23.46%)
Mutual labels: caffe
Uvw
Header-only, event based, tiny and easy to use libuv wrapper in modern C++ - now available as also shared/static library!
Stars: β 1,222 (+1408.64%)
Mutual labels: wrapper
Dispnet Flownet Docker
Dockerfile and runscripts for DispNet and FlowNet1 (estimation of disparity and optical flow)
Stars: β 78 (-3.7%)
Mutual labels: caffe
Domonit
A Deadly Simple Docker Monitoring Wrapper For Docker API
Stars: β 67 (-17.28%)
Mutual labels: wrapper
Hentai
Implements a wrapper class around nhentai's RESTful API.
Stars: β 68 (-16.05%)
Mutual labels: wrapper
Minecraft Wrapper
A simple & intuitive Minecraft Server wrapper. Supports IRC, backups, a plugin system, and more.
Stars: β 77 (-4.94%)
Mutual labels: wrapper
Pwc Net
PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)
Stars: β 1,142 (+1309.88%)
Mutual labels: caffe
Mobilenet Caffe
Caffe Implementation of Google's MobileNets (v1 and v2)
Stars: β 1,217 (+1402.47%)
Mutual labels: caffe
Csnackbar
This is a wrapper for android Snackbar. Which giving support to change Snackbar color, duration, message or even it's content view with a custom view.
Stars: β 76 (-6.17%)
Mutual labels: wrapper
React Native Line
Line SDK wrapper for React Native π
Stars: β 80 (-1.23%)
Mutual labels: wrapper
Pswritecolor
Write-Color is a wrapper around Write-Host allowing you to create nice looking scripts, with colorized output.
Stars: β 78 (-3.7%)
Mutual labels: wrapper
TRTWrapper
Desc
a wrapper for tensorRT net (parser caffe)
Test Environments
Ubuntu 16.04
TensorRT 5.0.2.6/4.0.1.6
CUDA 9.2
About Wraper
you can use the wrapper like this:
//normal
std::vector<std::vector<float>> calibratorData;
trtNet net("vgg16.prototxt","vgg16.caffemodel",{"prob"},calibratorData);
//fp16
trtNet net_fp16("vgg16.prototxt","vgg16.caffemodel",{"prob"},calibratorData,RUN_MODE:FLOAT16);
//int8
trtNet net_int8("vgg16.prototxt","vgg16.caffemodel",{"prob"},calibratorData,RUN_MODE:INT8);
//run inference:
net.doInference(input_data.get(), outputData.get());
//can print time cost
net.printTime();
//can write to engine and load From engine
net.saveEngine("save_1.engine");
trtNet net2("save_1.engine");
when you need add new plugin ,just add the plugin code to pluginFactory
Run Sample
#for classification
cd sample
mkdir build
cd build && cmake .. && make && make install
cd ..
./install/runNet --caffemodel=${CAFFE_MODEL_NAME} --prototxt=${CAFFE_PROTOTXT} --input=./test.jpg
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].