faster calibration curve generation
jellyfin intergration
This commit is contained in:
@@ -86,6 +86,16 @@ FetchContent_Declare(
|
||||
)
|
||||
FetchContent_MakeAvailable(nlohmann_json)
|
||||
|
||||
# nanobind (Python bindings for the sae_embed module)
|
||||
find_package(Python 3.8 COMPONENTS Interpreter Development.Module REQUIRED)
|
||||
FetchContent_Declare(
|
||||
nanobind
|
||||
GIT_REPOSITORY https://github.com/wjakob/nanobind.git
|
||||
GIT_TAG v2.4.0
|
||||
GIT_SHALLOW TRUE
|
||||
)
|
||||
FetchContent_MakeAvailable(nanobind)
|
||||
|
||||
# ── Model paths ───────────────────────────────────────────────────────────────
|
||||
set(SAE_MODELS_DIR "${CMAKE_SOURCE_DIR}/models"
|
||||
CACHE PATH "Directory containing ONNX model files")
|
||||
@@ -120,6 +130,15 @@ target_link_libraries(embed_faces PRIVATE
|
||||
)
|
||||
target_compile_definitions(embed_faces PRIVATE SAE_MODELS_DIR="${SAE_MODELS_DIR}")
|
||||
|
||||
# ── sae_embed — Python module: load SCRFD+ArcFace once, embed many images ───
|
||||
nanobind_add_module(sae_embed src/python_bindings.cpp)
|
||||
target_include_directories(sae_embed PRIVATE src)
|
||||
target_link_libraries(sae_embed PRIVATE
|
||||
${OpenCV_LIBS}
|
||||
onnxruntime
|
||||
)
|
||||
target_compile_definitions(sae_embed PRIVATE SAE_MODELS_DIR="${SAE_MODELS_DIR}")
|
||||
|
||||
# ── analyze — main analysis binary ───────────────────────────────────────────
|
||||
add_executable(scene_analyze src/main.cpp)
|
||||
target_link_libraries(scene_analyze PRIVATE sae_gallery)
|
||||
|
||||
Reference in New Issue
Block a user