Add DenseNet121 (wang-xinyu#460)

* add: densenet

* update readme

Author: makaveli10

densenet/CMakeLists.txt

@@ -0,0 +1,36 @@
+cmake_minimum_required(VERSION 3.10)
+
+# set the project name
+project(densenet)
+
+add_definitions(-std=c++11)
+
+# get main project dir to include common files
+get_filename_component(MAIN_DIR ../ ABSOLUTE)
+
+# When enabled the static version of the 
+# CUDA runtime library will be used in CUDA_LIBRARIES
+option(CUDA_USE_STATIC_CUDA_RUNTIME OFF)
+
+# specify the C++ standard
+set(CMAKE_CXX_STANDARD 11)
+set(CMAKE_CXX_STANDARD_REQUIRED True)
+set(CMAKE_BUILD_TYPE Debug)
+
+# include
+
+# include and link cuda
+include_directories(/usr/local/cuda/include)
+link_directories(/usr/local/cuda/lib64)
+
+# include and link tensorrt
+include_directories(/usr/include/x86_64-linux-gnu)
+link_directories(/usr/lib/x86_64-linux-gnu)
+
+# add the executable
+add_executable(densenet ${PROJECT_SOURCE_DIR}/densenet121.cpp)
+
+target_link_libraries(densenet nvinfer)
+target_link_libraries(densenet cudart)
+
+add_definitions(-O2 -pthread)

densenet/README.md

@@ -0,0 +1,38 @@
+# Densenet121
+
+The Pytorch implementation is [makaveli10/densenet](https://github.com/makaveli10/torchtrtz/tree/main/densenet).
+The tensorrt implemenation is taken from [makaveli10/cpptensorrtz](https://github.com/makaveli10/cpptensorrtz/).
+
+## How to Run
+
+1. generate densenet121.wts from pytorch
+
+```
+git clone https://github.com/wang-xinyu/tensorrtx.git
+git clone https://github.com/makaveli10/torchtrtz.git
+
+// go to torchtrtz/densenet
+// Enter these two commands to create densenet121.wts
+ $ python models.py
+ $ python gen_trtwts.py
+```
+
+2. build densenet and run
+
+```
+// put densenet121.wts into tensorrtx/densenet
+// go to tensorrtx/densenet
+mkdir build
+cd build
+cmake ..
+make
+sudo ./densenet -s  // serialize model to file i.e. 'LPRnet.engine'
+sudo ./densenet -d  // deserialize model and run inference
+```
+
+3. Verify output from [torch impl](https://github.com/makaveli10/torchtrtz/blob/main/densenet/README.md)
+TensorRT output[:5]:
+```
+    [-0.587389, -0.329202, -1.83404, -1.89935, -0.928404]
+```
+