Packages

evision

0.1.27
1.0.1-rc.0 1.0.0 1.0.0-rc.0 0.2.17 0.2.17-rc2 0.2.17-rc1 0.2.16 0.2.16-pre.2 0.2.16-pre 0.2.15 0.2.14 0.2.13 0.2.12 0.2.11 0.2.9 0.2.8 0.2.7 0.2.6 0.2.5 0.2.4 0.2.3 0.2.2 0.2.2-rc2 0.2.1 0.2.0 0.1.39 0.1.38 0.1.37 0.1.36 0.1.35 0.1.34 0.1.33 0.1.32 retired 0.1.31 0.1.30 0.1.29 0.1.28 0.1.27 0.1.26 0.1.26-rc3 0.1.26-rc2 0.1.26-rc1 0.1.26-rc0 0.1.25 0.1.24 0.1.23 0.1.22 0.1.21 0.1.20 0.1.19 0.1.18 0.1.17 0.1.16 0.1.15 0.1.14 0.1.13 0.1.12 0.1.11 0.1.10 0.1.9 0.1.8 0.1.7

OpenCV-Erlang/Elixir binding.

HexDocs

Current section

Files

Jump to
Readme
62 Versions
6 Dependencies
Dependants
Files
Activity
evision patches apply_patch.py
Raw

patches/apply_patch.py 

#!/usr/bin/env python3

# -*- coding: utf-8 -*-



import sys

from pathlib import Path

if sys.version_info[0] >= 3:

    from io import StringIO

else:

    from cStringIO import StringIO





def patch_fix_getLayerShapes(opencv_version: str, opencv_src_root: str):

    if opencv_version not in ['4.5.4', '4.5.5', '4.6.0', '4.7.0']:

        print(f"warning: applying `patch_fix_getLayerShapes` to opencv version `{opencv_version}`")



    # modules/dnn/include/opencv2/dnn/dnn.hpp

    dnn_hpp = Path(opencv_src_root) / 'modules' / 'dnn' / 'include' / 'opencv2' / 'dnn' / 'dnn.hpp'

    fixed = StringIO()

    patched_1 = False

    patched_2 = False

    with open(dnn_hpp, 'r') as source:

        for line in source:

            if not patched_1 and line.strip() == 'void getLayerShapes(const MatShape& netInputShape,':

                fixed.write('        CV_WRAP void getLayerShapes(const MatShape& netInputShape,\n')

                patched_1 = True

            elif not patched_2 and line.strip() == 'void getLayerShapes(const std::vector<MatShape>& netInputShapes,':

                fixed.write('        CV_WRAP void getLayerShapes(const std::vector<MatShape>& netInputShapes,\n')

                patched_2 = True

            else:

                fixed.write(line)



    if patched_1 or patched_2:

        with open(dnn_hpp, 'w') as dst:

            dst.truncate(0)

            dst.write(fixed.getvalue())





def patch_winograd(opencv_version: str, opencv_src_root: str):

    if opencv_version not in ['4.7.0']:

        print(f"warning: skipped applying `patch_winograd` to opencv version `{opencv_version}`")

        return



    # modules/dnn/src/layers/fast_convolution/fast_convolution.cpp

    fast_convolution_cpp = Path(opencv_src_root) / 'modules' / 'dnn' / 'src' / 'layers' / 'fast_convolution' / 'fast_convolution.cpp'

    fixed = StringIO()

    patched_1 = False

    with open(fast_convolution_cpp, 'r') as source:

        for line in source:

            if not patched_1 and line.strip() == 'Mat weightsMat = _weightsMat.getMat();':

                fixed.write("""#if CV_TRY_AVX2

    // Disabel Winograd when CV_TRY_AVX2 is true, but conv->useAVX2 is false.

    if (conv->conv_type == _FX_CONV_TYPE_WINOGRAD3X3 && !conv->useAVX2)

        conv->conv_type = _FX_CONV_TYPE_GENERIC;

#endif



    Mat weightsMat = _weightsMat.getMat(); // patched\n""")

                patched_1 = True

            else:

                fixed.write(line)



    if patched_1:

        with open(fast_convolution_cpp, 'w') as dst:

            dst.truncate(0)

            dst.write(fixed.getvalue())





def patch_rpath_linux(opencv_version: str, opencv_src_root: str):

    # CMakeLists.txt

    cmakelists_txt = Path(opencv_src_root) / 'CMakeLists.txt'

    fixed = StringIO()

    patched_1 = False

    with open(cmakelists_txt, 'r') as source:

        for line in source:

            if not patched_1 and line.strip() == 'string(REPLACE "opencv_" "" OPENCV_MODULES_BUILD_ST          "${OPENCV_MODULES_BUILD_ST}")':

                fixed.write("""if(UNIX AND NOT APPLE)

  foreach(the_module ${OPENCV_MODULES_BUILD_ST})

    set_target_properties(${the_module} PROPERTIES

      INSTALL_RPATH_USE_LINK_PATH TRUE

      BUILD_WITH_INSTALL_RPATH TRUE

    )

    set_target_properties(${the_module} PROPERTIES INSTALL_RPATH "\$ORIGIN")

  endforeach()

endif()

string(REPLACE "opencv_" "" OPENCV_MODULES_BUILD_ST          "${OPENCV_MODULES_BUILD_ST}") # patched

""")

                patched_1 = True

            else:

                fixed.write(line)



    if patched_1:

        with open(cmakelists_txt, 'w') as dst:

            dst.truncate(0)

            dst.write(fixed.getvalue())





patches = [patch_fix_getLayerShapes, patch_winograd, patch_rpath_linux]





if __name__ == '__main__':

    cv_version = None

    cv_src_root = None

    if len(sys.argv) != 3:

        sys.exit(1)

    cv_src_root = sys.argv[1]

    cv_version = sys.argv[2]

    print(f"[+] applying patches to OpenCV {cv_version} at {cv_src_root}")

    for p in patches:

        p(cv_version, cv_src_root)