ctpn:Run demo.py on windows with cpu successfully!

清疚 2023-08-17 17:07 71阅读 0赞

my environment is:
    windows10 ,
    python3.6 ,
    tensorflow1.3 ,
    vs2015(ps:vs2013 not support python3.6 when compile)
    
    step 1:make some change
    change "np.int_t " to "np.intp_t" in line 25 of the file lib\utils\cython_nms.pyx
    otherwise appear " ValueError: Buffer dtype mismatch, expected 'int_t' but got 'long long' " in step 6.
    
    step 2:updata c file
    execute:cd your_dir\text-detection-ctpn-master\lib\utils
    execute:cython bbox.pyx
    execute:cython cython_nms.pyx
    
    step 3:builf setup file as setup_new.py
    import numpy as np
    from distutils.core import setup
    from Cython.Build import cythonize
    from distutils.extension import Extension
    numpy_include = np.get_include()
    setup(ext_modules=cythonize("bbox.pyx"),include_dirs=[numpy_include])
    setup(ext_modules=cythonize("cython_nms.pyx"),include_dirs=[numpy_include])
    
    step 4:build .pyd file
    execute:python setup_new.py install
    copy bbox.cp36-win_amd64.pyd and cython_nms.cp36-win_amd64.pyd to your_dir\text-detection-ctpn-master\lib\utils
    
    step 5:make some change
    (1) Set "USE_GPU_NMS " in the file \ctpn\text.yml as "False"
    (2) Set the "_C.USE_GPU_NMS" in the file \lib\fast_rcnn\config.py as "False";
    (3) Comment out the line "from lib.utils.gpu_nms import gpu_nms" in the file \lib\fast_rcnn\nms_wrapper.py;
    (4) Comment out the line "from . import gpu_nms" in the file \lib\utils_init.py;
    (5) change "base_name = image_name.split('/')[-1]" to "base_name = image_name.split('\')[-1]" in line 24 of the file ctpn\demo.py
    
    step 6:run demo
    execute:cd your_dir\text-detection-ctpn-master
    execute:python ./ctpn/demo.py