Pyinstaller 打包exe运行时找不到源码，函数 错误 OSError: could not get source code

今天用pyinstaller将Detectron2的一个文件打包成了exe，可是在程序运行过程中出现了下面错误

Traceback (most recent call last):
  File "torch_sources.py", line 21, in get_source_lines_and_file
    sourcelines, file_lineno = inspect.getsourcelines(obj)
  File "inspect.py", line 956, in getsourcelines
  File "inspect.py", line 787, in findsource
OSError: could not get source code

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File "cell_analyse.py", line 8, in 
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "detectron2model_zoo__init__.py", line 8, in 
    from .model_zoo import get, get_config_file, get_checkpoint_url, get_config
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "detectron2model_zoomodel_zoo.py", line 9, in 
    from detectron2.modeling import build_model
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "detectron2modeling__init__.py", line 2, in 
    from detectron2.layers import ShapeSpec
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "detectron2layers__init__.py", line 2, in 
    from .batch_norm import FrozenBatchNorm2d, get_norm, NaiveSyncBatchNorm, CycleBatchNormList
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "detectron2layersbatch_norm.py", line 4, in 
    from fvcore.nn.distributed import differentiable_all_reduce
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "fvcorenn__init__.py", line 4, in 
  File "", line 983, in _find_and_load
  File "", line 967, in _find_and_load_unlocked
  File "", line 677, in _load_unlocked
  File "PyInstallerloaderpyimod03_importers.py", line 495, in exec_module
  File "fvcorennfocal_loss.py", line 52, in 
  File "torchjit_script.py", line 1307, in script
    ast = get_jit_def(obj, obj.__name__)
  File "torchjitfrontend.py", line 233, in get_jit_def
    parsed_def = parse_def(fn)
  File "torch_sources.py", line 95, in parse_def
    sourcelines, file_lineno, filename = get_source_lines_and_file(fn, ErrorReport.call_stack())
  File "torch_sources.py", line 28, in get_source_lines_and_file
    raise OSError(msg) from e
OSError: Can't get source for . TorchScript requires source access in order to carry out compilation, make sure original .py files are available.
[75064] Failed to execute script 'cell_analyse' due to unhandled exception!

这个错误是直接提示为sigmoid_focal_loss 这个函数的源代码没有找到，而添加源代码的程序是File “inspect.py”, line 787, in findsource，进一步追踪代码发现可以通过getfile这个函数可以找到程序具体要添加哪些模块，我在这个函数的代码里面添加了下面这行，打印出具体是哪个模块没有添加。

def getfile(object):
    """Work out which source or compiled file an object was defined in."""
    if ismodule(object):
        if getattr(object, '__file__', None):
            return object.__file__
        raise TypeError('{!r} is a built-in module'.format(object))
    if isclass(object):
        if hasattr(object, '__module__'):
            object = sys.modules.get(object.__module__)
            if getattr(object, '__file__', None):
                return object.__file__
        raise TypeError('{!r} is a built-in class'.format(object))
    if ismethod(object):
        object = object.__func__
    if isfunction(object):
        object = object.__code__
        **print(object)**     **#在这里添加这行代码，打印出到底是哪些源代码没有添加成功**
    if istraceback(object):
        object = object.tb_frame
    if isframe(object):
        object = object.f_code
    if iscode(object):
        return object.co_filename
    raise TypeError('module, class, method, function, traceback, frame, or '
                    'code object was expected, got {}'.format(
                    type(object).__name__))

个人感觉这个坑爹的错误是源代码加载有关的，Detectron2使用了inspect这个模块，加载源代码，导致Pyinstaller在分析文件依赖时，没有把相关的源代码添加进去。
添加好打印代码后，再次编译生成exe，重新运行，产生如下log

Traceback (most recent call last):
  File "torch_sources.py", line 21, in get_source_lines_and_file
    sourcelines, file_lineno = inspect.getsourcelines(obj)
  File "inspect.py", line 956, in getsourcelines
  File "inspect.py", line 787, in findsource
OSError: could not get source code

由此可以看出来，是fvcore这个模块没有被添加到可行exe目录下。当我把源代码库下的fvcore拷贝到生成exe所在的目录后，再次运行，就不在报fvcore找不到错误了，可以报了其它模块源代码找不到错误，我以此拷贝过去，程序便可以顺利运行了。

当然正规的解决办法应该是通过修改.spec 文件，将需要依赖的文件在打包时自动添加到编译生成的文件里面，这一步，可以参考pyinstaller的官方解决方法