这是一个即插即用的模块,自适应不同大小的输入(不需要手动输入任何参数)
if __name__ == '__main__':中的代码展示了对不同size张量的自适应性
在模型的类定义中,按以下方式处理:
- from OIPModule.py import OIPModule
- 在__init__函数中,加入
self.oip = OIPModule() - 在forward函数中,加入
map = self.oip(map)即可完成模块插入
数据集与下载链接:
| Name | Name | Last commit date | ||
|---|---|---|---|---|