add resnet34 (wang-xinyu#741)

* hello

* add resnet34

Author: liufqing

resnet/CMakeLists.txt

@@ -21,6 +21,10 @@ add_executable(resnet18 ${PROJECT_SOURCE_DIR}/resnet18.cpp)
 target_link_libraries(resnet18 nvinfer)
 target_link_libraries(resnet18 cudart)
 
+add_executable(resnet34 ${PROJECT_SOURCE_DIR}/resnet34.cpp)
+target_link_libraries(resnet34 nvinfer)
+target_link_libraries(resnet34 cudart)
+
 add_executable(resnet50 ${PROJECT_SOURCE_DIR}/resnet50.cpp)
 target_link_libraries(resnet50 nvinfer)
 target_link_libraries(resnet50 cudart)

resnet/README.md

@@ -13,11 +13,11 @@ Following tricks are used in this resnet, nothing special, residual connection a
 ## TensorRT C++ API
 
 ```
-// 1a. generate resnet18.wts or resnet50.wts from [pytorchx/resnet](https://github.com/wang-xinyu/pytorchx/tree/master/resnet)
+// 1a. generate resnet18.wts,resnet34.wts or resnet50.wts from [pytorchx/resnet](https://github.com/wang-xinyu/pytorchx/tree/master/resnet)
 
 // 1b. generate wide_resnet50.wts from [BlueMirrors/torchtrtz](https://github.com/BlueMirrors/torchtrtz)
 
-// 2. put resnet18.wts or resnet50.wts into tensorrtx/resnet
+// 2. put resnet18.wts,resnet34 or resnet50.wts into tensorrtx/resnet
 
 // 3. build and run
 
@@ -34,6 +34,10 @@ make
 sudo ./resnet18 -s   // serialize model to plan file i.e. 'resnet18.engine'
 sudo ./resnet18 -d   // deserialize plan file and run inference
 
+or
+sudo ./resnet34 -s   // serialize model to plan file i.e. 'resnet34.engine'
+sudo ./resnet34 -d   // deserialize plan file and run inference
+
 or
 
 sudo ./resnet50 -s   // serialize model to plan file i.e. 'resnet50.engine'
@@ -51,7 +55,7 @@ sudo ./wide_resnet50 -d   // deserialize plan file and run inference
 
 
 // 4. see if the output is same as 
-- [pytorchx/resnet](https://github.com/wang-xinyu/pytorchx/tree/master/resnet) - for resnet18, resnet50, resnext50
+- [pytorchx/resnet](https://github.com/wang-xinyu/pytorchx/tree/master/resnet) - for resnet18, resnet34, resnet50, resnext50
 - [BlueMirrors/torchtrtz](https://github.com/BlueMirrors/torchtrtz) - for wide_resnet50
 ```