# Check OpenCV
if(NOT USE_OPENCV OR NOT OpenCV_FOUND OR OpenCV_VERSION_MAJOR LESS 3)
  message(WARNING "\
OpenCV version >= 3 should be enabled and found to build image classification example, skipping...")
  return()
endif()

if(NOT MSVC)
  set(IMG_CLASSIFICATION_EXAMPLE_STATIC_LINK ON CACHE BOOL "\
Link mxnet library statically in the c++ image classification example")
else()
  # disable static linking on Windows
  set(IMG_CLASSIFICATION_EXAMPLE_STATIC_LINK OFF)
endif()

add_executable(image-classification-predict image-classification-predict.cc)
include_directories(SYSTEM ${OpenCV_INCLUDE_DIRS})

if(IMG_CLASSIFICATION_EXAMPLE_STATIC_LINK)
  target_link_libraries(image-classification-predict
                        ${BEGIN_WHOLE_ARCHIVE} mxnet_static ${END_WHOLE_ARCHIVE}
                        dmlc
                        ${mxnet_LINKER_LIBS}
                        )
  add_dependencies(image-classification-predict mxnet_static)
else()
  target_link_libraries(image-classification-predict
                        dmlc
                        ${nnvm_LINKER_LIBS}
                        ${mxnet_LINKER_LIBS}
                        mxnet
                        )
  add_dependencies(image-classification-predict mxnet)
endif()


