原因分析
如上图,当在命令行中执行以下命令时:
1 | python cls_train.py |
会提示找不到datasets
模块,原因是在命令行运行python文件,系统只会在当前文件夹runs/
下寻找,不会到上一级目录查找,在pycharm等IDE中会自动在整个项目中查找,所以命令行运行python文件会提示找不到包。
解决方法
在cls_train.py
文件中,添加以下代码,将项目根目录加入搜索路径即可,一定要写在导入其他模块之前
1 | import sys |
见下图:
现在再运行python cls_train.py
就没问题了。