cmake_minimum_required(VERSION 3.5)
project(hello_trt)
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/lib)
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/lib)
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/bin)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++14")
SET(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS};-gencode arch=compute_61,code=sm_61;)
message(STATUS "CMAKE_RUNTIME_OUTPUT_DIRECTORY: "${CMAKE_RUNTIME_OUTPUT_DIRECTORY})
set(3RDPARTY ${CMAKE_CURRENT_SOURCE_DIR}/3rd)
set(TENSORRT_LIB ${3RDPARTY}/TensorRT-8.0.1.6/targets/x86_64-linux-gnu/lib)
file(GLOB TENSORRT_LIBS "${TENSORRT_LIB}/*.so")
set(OpenCV_DIR ${3RDPARTY}/opencv-3.4.16/share/OpenCV)
find_package(CUDA REQUIRED)
find_package(OpenCV REQUIRED)
message(STATUS "CUDA_LIBRARIES: "${CUDA_LIBRARIES})
message(STATUS "TENSORRT_LIBS: "${TENSORRT_LIBS})
message(STATUS "Opencv_LIBS: "${OpenCV_LIBS})
include_directories(
${CUDA_INCLUDE_DIRS}
${3RDPARTY}/TensorRT-8.0.1.6/include
${CMAKE_CURRENT_SOURCE_DIR}/include
${OpenCV_INCLUDE_DIRS}
)
add_executable(
trt_cls
src/TrtCls.cpp
)
target_link_libraries(
trt_cls
${CUDA_LIBRARIES}
${TENSORRT_LIBS}
${OpenCV_LIBS}
)
基于torchision的图像分类深度学习模型.zip
需积分: 0 31 浏览量
2023-10-10
22:43:16
上传
评论 1
收藏 3.03MB ZIP 举报
pytorch_image_classification.zip （50个子文件） 
pytorch_image_classification tools 
evaluation.py 5KB train_val.py 6KB predict.py 2KB sample_files imgs dog 
2.jpg 15KB 1.jpg 26KB 5.jpg 39KB 0.jpg 34KB 3.jpg 30KB 4.jpg 106KB cat 2.jpg 26KB 1.jpg 16KB 5.jpg 6KB 0.jpg 12KB 3.jpg 37KB 4.jpg 20KB
listfile.txt 804B cpp_inference traced_model trace_model.py 711B compile.sh 103B CMakeLists.txt 875B .vscode settings.json 1KB classification include ImgCls.hpp 700B CMakeLists.txt 769B src ImgCls.cpp 2KB test_ImgCls.cpp 511B utils __init__.py 155B util.py 2KB loss_kd.py 621B lr_scheduler.py 3KB warmup_lr.py 4KB label_smoothing_pytorch.py 1KB opts.py 3KB dataset __init__.py 90B cls_dataset.py 1KB transform.py 2KB visualization 
test.png 143KB f1_conv1.png 1.51MB Feature_Visualization.py 3KB Feature_Visualization.ipynb 1.73MB trt_inference include TrtCls.hpp 4KB CMakeLists.txt 1KB src TrtCls.cpp 8KB convert_onnx.py 2KB labels.txt 12B cls_models __init__.py 27B model.py 3KB ensamble kaggle_vote.py 2KB samples method_3.csv 83KB method_2.csv 83KB method_1.csv 83KB run.sh 1KB资源评论
