今天想给模型换一个backbone时,突然发现不知道怎么导入不同目录下的包,具体是这样的,如下图所示:
现在我们工作在change_backbone_without_fpn.py
文件中,希望导入swin_transformer_model.py
中的模型。【注意change_backbone_without_fpn.py
和swin_transformer_model.py
不在一个文件夹下,swin_transformer_model.py
上级的backbone和change_backbone_without_fpn.py
在同一个文件夹中】
这该怎么导入呢?直接来说方法,可以在拟导包所在的文件夹下,即backbone文件夹下新建一个__init__.py
文件,可以不写任何内容,然后在change_backbone_without_fpn.py
文件中使用from backbone.swin_transformer_model import *
导入该文件下所有模块,如果想导入某个具体的模块,只需将*替换成对应模块名即可。
其实啊,我们也可以在__init__.py
文件中写入一些内容,如写入from .swin_transformer_model import *
,这样写了后,我们只需要在change_backbone_without_fpn.py
文件中写入from backbone import *
即可实现导入所有模块,其实这两种方式差不多啦,大家看别人代码的时候注意一下这个__init__.py
就好,否则你可能不知道为什么别人有的包能够导入成功,而你不行!!!
🌷🌷🌷🌷🌷🌷🌷
2022.10.28更新
上文介绍的目录结构类似是下图这样的:
├──root
├──文件夹1
├──test1
├──test2
即将test1中的包导入到test2中。
然而对于下面的目录结构上述方法似乎就不行了:
├──root
├──文件夹1
├──test1
├──文件夹2
├──test2
解决方案也很简单,代码如下:
import sys
sys.path.append("..") ## 指定到上一级目录
from 文件夹1.test1 import *
是不是挺简单的呢,快去试试吧!!!🌱🌱🌱
🌷🌷🌷🌷🌷🌷🌷
这篇博文主要是记录一下自己导包时遇到的小麻烦,如果能帮到大家就更好了,希望大家学习过程中都能有所收获喔🥗🥗🥗