Skip to content

Commit 917eab4

Browse files
torch yolov7
1 parent 550ae66 commit 917eab4

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ endif(DLib)
4141

4242
if(torch)
4343
#set(CMAKE_PREFIX_PATH "${CMAKE_PREFIX_PATH}/home/anton/Libs/libtorch")
44-
list(APPEND CMAKE_PREFIX_PATH "/home/anton/Libs/libtorch")
44+
list(APPEND CMAKE_PREFIX_PATH "/home/anton/Lib/libtorch")
4545
message(${CMAKE_PREFIX_PATH})
4646
find_package(Torch REQUIRED)
4747
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
@@ -130,7 +130,7 @@ set( EOD_SRC
130130
src/lib/detectors/ExtractedInfoStringChecker.cpp
131131
src/lib/detectors/UnitTranslationExtracter.cpp
132132
src/lib/detectors/SquareObjectDistanceDetector.cpp
133-
src/lib/detectors/TorchDetector.cpp
133+
src/lib/detectors/TorchYOLOv7Detector.cpp
134134

135135
# lib relations
136136
src/lib/relations/ImageRangeRelation.cpp

0 commit comments

Comments
 (0)