cmake_minimum_required (VERSION 3.8)
project(SOLDIER)
set(Torch_DIR "/libtorch/share/cmake/Torch")
set(PYTHON_EXECUTABLE "/usr/bin/python3")
find_package(Torch REQUIRED)
find_package(OpenCV REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
set(CMAKE_BUILD_TYPE Debug)
include_directories(${CMAKE_SOURCE_DIR}/include)
include_directories(${OpenCV_INCLUDE_DIRS})
add_executable(run src/main.cpp src/cnn.cpp src/dataset.cpp include/cnn.h include/dataset.h)
target_link_libraries(run "${OpenCV_LIBS}" "${TORCH_LIBRARIES}")
set_property(TARGET run PROPERTY CXX_STANDARD 14)
c++编写的CNN算法内含数据集以及训练预测源码
版权申诉
169 浏览量
2022-10-21
22:35:24
上传
评论
收藏 12KB ZIP 举报
307Cnn.zip （13个子文件） 
307Cnn include 
cnn.h 1018B dataset.h 941B src dataset.cpp 2KB main.cpp 5KB cnn.cpp 967B
CMakeLists.txt 580B .vscode launch.json 509B targets.log 7KB settings.json 2KB tasks.json 684B c_cpp_properties.json 470B configurationCache.log 139B dryrun.log 253B资源评论
