ChAI/demos/video/CMakeLists.txt at 61cdba455bfd459684328528d3b950734dfde3b1 · Iainmon/ChAI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97


# export OpenCV_DIR="$(brew --prefix opencv)/share/opencv4"
# /opt/homebrew/opt/opencv/share/opencv4
find_package(OpenCV 4 REQUIRED)

find_library(ACCELERATE Accelerate REQUIRED)
find_library(METAL Metal REQUIRED)
find_library(FOUNDATION Foundation REQUIRED)


add_executable(VidStreamer
    ${CMAKE_CURRENT_SOURCE_DIR}/webcam-capture/webcam_infer.cpp
)

target_include_directories(VidStreamer
    PRIVATE
        ${CMAKE_CURRENT_SOURCE_DIR}/include
        ${LIBTORCH_DIR}/include
        ${LIBTORCH_DIR}/include/torch/csrc/api/include
)

target_link_directories(VidStreamer PRIVATE ${LIBTORCH_DIR}/lib)

target_link_libraries(VidStreamer
    PRIVATE
        -ltorch
        -ltorch_cpu
        -lc10
        -ltorch_global_deps
        ${OpenCV_LIBS}
        # ${TORCH_LIBRARIES}
        ${ACCELERATE}
        ${METAL}
        ${FOUNDATION}
)

set_target_properties(VidStreamer PROPERTIES
    RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}
)

if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
  target_compile_options(VidStreamer PRIVATE -Ofast -flto -ffast-math)
  target_link_options(VidStreamer PRIVATE -flto)
endif()


add_executable(StyleTransfer
    ${CMAKE_CURRENT_SOURCE_DIR}/style-transfer/style_transfer.cpp
)

target_include_directories(StyleTransfer
    PRIVATE
        ${CMAKE_CURRENT_SOURCE_DIR}/include
        ${LIBTORCH_DIR}/include
        ${LIBTORCH_DIR}/include/torch/csrc/api/include
)

target_link_directories(StyleTransfer PRIVATE ${LIBTORCH_DIR}/lib)

target_link_libraries(StyleTransfer
    PRIVATE
        -ltorch
        -ltorch_cpu
        -lc10
        -ltorch_global_deps
        ${OpenCV_LIBS}
        # ${TORCH_LIBRARIES}
        ${ACCELERATE}
        ${METAL}
        ${FOUNDATION}
)

set_target_properties(StyleTransfer PROPERTIES
    RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}
)

if (CMAKE_CXX_COMPILER_ID MATCHES "Clang")
  target_compile_options(StyleTransfer PRIVATE -Ofast -flto -ffast-math)
  target_link_options(StyleTransfer PRIVATE -flto)
endif()


add_custom_command(
    TARGET StyleTransfer
    POST_BUILD
    COMMAND ${CMAKE_COMMAND} -E copy_directory
            "${CMAKE_CURRENT_SOURCE_DIR}/style-transfer/models"
            "$<TARGET_FILE_DIR:StyleTransfer>/style-transfer/models"
    COMMENT "NOT! Copying ${PROJECT_ROOT_DIR}/examples/vgg/images to $<TARGET_FILE_DIR:vgg>/images"
)