File size: 4,263 Bytes
adc9971 |
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 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 |
// Copyright (C) 2015 Davis E. King (davis@dlib.net)
// License: Boost Software License See LICENSE.txt for the full license.
#include "opaque_types.h"
#include <pybind11/pybind11.h>
#include <dlib/simd.h>
#include <string>
namespace py = pybind11;
void bind_matrix(py::module& m);
void bind_vector(py::module& m);
void bind_svm_c_trainer(py::module& m);
void bind_decision_functions(py::module& m);
void bind_basic_types(py::module& m);
void bind_other(py::module& m);
void bind_svm_rank_trainer(py::module& m);
void bind_cca(py::module& m);
void bind_sequence_segmenter(py::module& m);
void bind_svm_struct(py::module& m);
void bind_image_classes(py::module& m);
void bind_image_classes2(py::module& m);
void bind_image_classes3(py::module& m);
void bind_image_classes4(py::module& m);
void bind_rectangles(py::module& m);
void bind_object_detection(py::module& m);
void bind_shape_predictors(py::module& m);
void bind_correlation_tracker(py::module& m);
void bind_face_recognition(py::module& m);
void bind_cnn_face_detection(py::module& m);
void bind_global_optimization(py::module& m);
void bind_numpy_returns(py::module& m);
void bind_image_dataset_metadata(py::module& m);
void bind_line(py::module& m);
#ifndef DLIB_NO_GUI_SUPPORT
void bind_gui(py::module& m);
#endif
PYBIND11_MODULE(_dlib_pybind11, m)
{
// Attempt to give users that have compiled dlib to use SIMD instructions but then tried to use
// it on a CPU that doesn't support them a more informative warning that many systems generate
// by default in such events. Note that this may or may not be able to print a warning before
// some unavailable SIMD instruction is used. It depends on what SIMD instructions your
// compiler may or may not have inserted into other parts of the code that it compiles. Which
// is entirely up to your compiler. And by the nature of this user error, we can't control how
// this plays out. So on some systems/setups/compilers they will get this nicer error message
// and sometimes they don't.
warn_about_unavailable_but_used_cpu_instructions();
#define DLIB_QUOTE_STRING(x) DLIB_QUOTE_STRING2(x)
#define DLIB_QUOTE_STRING2(x) #x
m.attr("__version__") = DLIB_QUOTE_STRING(DLIB_VERSION);
m.attr("__time_compiled__") = std::string(__DATE__) + " " + std::string(__TIME__);
#ifdef DLIB_USE_CUDA
m.attr("DLIB_USE_CUDA") = true;
#else
m.attr("DLIB_USE_CUDA") = false;
#endif
#ifdef DLIB_USE_BLAS
m.attr("DLIB_USE_BLAS") = true;
#else
m.attr("DLIB_USE_BLAS") = false;
#endif
#ifdef DLIB_USE_LAPACK
m.attr("DLIB_USE_LAPACK") = true;
#else
m.attr("DLIB_USE_LAPACK") = false;
#endif
#ifdef DLIB_HAVE_AVX
m.attr("USE_AVX_INSTRUCTIONS") = true;
#else
m.attr("USE_AVX_INSTRUCTIONS") = false;
#endif
#ifdef DLIB_HAVE_NEON
m.attr("USE_NEON_INSTRUCTIONS") = true;
#else
m.attr("USE_NEON_INSTRUCTIONS") = false;
#endif
// Note that the order here matters. We need to do the basic types first. If we don't
// then what happens is the documentation created by sphinx will use horrible big
// template names to refer to C++ objects rather than the python names python users
// will expect. For instance, if bind_basic_types() isn't called early then when
// routines take a std::vector<double>, rather than saying dlib.array in the python
// docs it will say "std::vector<double, std::allocator<double> >" which is awful and
// confusing to python users.
//
// So when adding new things always add them to the end of the list.
bind_matrix(m);
bind_vector(m);
bind_basic_types(m);
bind_other(m);
bind_line(m);
bind_svm_rank_trainer(m);
bind_decision_functions(m);
bind_cca(m);
bind_sequence_segmenter(m);
bind_svm_struct(m);
bind_rectangles(m);
bind_image_classes(m);
bind_image_classes2(m);
bind_image_classes3(m);
bind_image_classes4(m);
bind_object_detection(m);
bind_shape_predictors(m);
bind_correlation_tracker(m);
bind_face_recognition(m);
bind_cnn_face_detection(m);
bind_global_optimization(m);
bind_numpy_returns(m);
bind_svm_c_trainer(m);
#ifndef DLIB_NO_GUI_SUPPORT
bind_gui(m);
#endif
bind_image_dataset_metadata(m);
}
|