今天想给模型换一个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
就好,否则你可能不知道为什么别人有的包能够导入成功,而你不行!!!
这篇博文主要是自己记录一下,如果能帮到大家就更好了,希望大家学习过程中都能有所收获喔