python 添加 import 的路径

  • 有时候我们会遇到在一份代码中想要用另一份代码的类或方法,比如深度学习想用到另一份代码里面写好的模型时,需要直接import另一个路径的类,有两种方法:

通过sh脚本来跑代码

比如想跑train.py,那么可以写一个train.sh,内容如下:

#!/usr/bin/env bash
PYTHONPATH="/path/to/mymodel":$PYTHONPATH \
python train.py

这个脚本的作用是在运行train.py时,将 /path/to/mymodel 这个路径添加到PYTHONPATH 变量中,这样的话import 模块时,也会到这个路径下去搜索可import 模块

通过sys模块添加路径

可以在train.py中添加这一行代码:

sys.path.append('/path/to/mymodel')

能够实现一样的效果

你可能感兴趣的:(实用代码,python)