├── .gitignore ├── CMakeLists.txt ├── README.md ├── cmake_install.cmake ├── common.cpp ├── common.h ├── cudaUtility.h ├── imageBuffer.h ├── kernel.cu ├── main.cpp ├── mathFunctions.cpp ├── mathFunctions.cu ├── mathFunctions.h ├── model └── pelee │ ├── pelee_deploy_iplugin.prototxt │ ├── pelee_merged.caffemodel │ └── pelee_merged.caffemodel.1.tensorcache ├── pluginImplement.cpp ├── pluginImplement.h ├── tensorNet.cpp ├── tensorNet.h ├── testPic └── test.png ├── testVideo └── test.avi └── util ├── cuda ├── cudaMappedMemory.h ├── cudaNormalize.cu ├── cudaNormalize.h ├── cudaOverlay.cu ├── cudaOverlay.h ├── cudaRGB.cu ├── cudaRGB.h ├── cudaResize.cu ├── cudaResize.h ├── cudaUtility.h ├── cudaYUV-NV12.cu ├── cudaYUV-YUYV.cu ├── cudaYUV-YV12.cu └── cudaYUV.h ├── loadImage.cpp └── loadImage.h /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/README.md -------------------------------------------------------------------------------- /cmake_install.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/cmake_install.cmake -------------------------------------------------------------------------------- /common.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/common.cpp -------------------------------------------------------------------------------- /common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/common.h -------------------------------------------------------------------------------- /cudaUtility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/cudaUtility.h -------------------------------------------------------------------------------- /imageBuffer.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/imageBuffer.h -------------------------------------------------------------------------------- /kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/kernel.cu -------------------------------------------------------------------------------- /main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/main.cpp -------------------------------------------------------------------------------- /mathFunctions.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/mathFunctions.cpp -------------------------------------------------------------------------------- /mathFunctions.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/mathFunctions.cu -------------------------------------------------------------------------------- /mathFunctions.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/mathFunctions.h -------------------------------------------------------------------------------- /model/pelee/pelee_deploy_iplugin.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/model/pelee/pelee_deploy_iplugin.prototxt -------------------------------------------------------------------------------- /model/pelee/pelee_merged.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/model/pelee/pelee_merged.caffemodel -------------------------------------------------------------------------------- /model/pelee/pelee_merged.caffemodel.1.tensorcache: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/model/pelee/pelee_merged.caffemodel.1.tensorcache -------------------------------------------------------------------------------- /pluginImplement.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/pluginImplement.cpp -------------------------------------------------------------------------------- /pluginImplement.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/pluginImplement.h -------------------------------------------------------------------------------- /tensorNet.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/tensorNet.cpp -------------------------------------------------------------------------------- /tensorNet.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/tensorNet.h -------------------------------------------------------------------------------- /testPic/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/testPic/test.png -------------------------------------------------------------------------------- /testVideo/test.avi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/testVideo/test.avi -------------------------------------------------------------------------------- /util/cuda/cudaMappedMemory.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaMappedMemory.h -------------------------------------------------------------------------------- /util/cuda/cudaNormalize.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaNormalize.cu -------------------------------------------------------------------------------- /util/cuda/cudaNormalize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaNormalize.h -------------------------------------------------------------------------------- /util/cuda/cudaOverlay.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaOverlay.cu -------------------------------------------------------------------------------- /util/cuda/cudaOverlay.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaOverlay.h -------------------------------------------------------------------------------- /util/cuda/cudaRGB.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaRGB.cu -------------------------------------------------------------------------------- /util/cuda/cudaRGB.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaRGB.h -------------------------------------------------------------------------------- /util/cuda/cudaResize.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaResize.cu -------------------------------------------------------------------------------- /util/cuda/cudaResize.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaResize.h -------------------------------------------------------------------------------- /util/cuda/cudaUtility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaUtility.h -------------------------------------------------------------------------------- /util/cuda/cudaYUV-NV12.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaYUV-NV12.cu -------------------------------------------------------------------------------- /util/cuda/cudaYUV-YUYV.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaYUV-YUYV.cu -------------------------------------------------------------------------------- /util/cuda/cudaYUV-YV12.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaYUV-YV12.cu -------------------------------------------------------------------------------- /util/cuda/cudaYUV.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/cuda/cudaYUV.h -------------------------------------------------------------------------------- /util/loadImage.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/loadImage.cpp -------------------------------------------------------------------------------- /util/loadImage.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cathy-kim/Pelee-TensorRT/HEAD/util/loadImage.h --------------------------------------------------------------------------------