Packages

evision

0.1.23
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']:

        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)



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

        dst.truncate(0)

        dst.write(fixed.getvalue())





patches = [patch_fix_getLayerShapes]





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(cv_version, cv_src_root)

    for p in patches:

        p(cv_version, cv_src_root)