Skip to content

Mantee0810/OIP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 

Repository files navigation

OIP

这是一个即插即用的模块,自适应不同大小的输入(不需要手动输入任何参数)

if __name__ == '__main__':

中的代码展示了对不同size张量的自适应性

用法

在模型的类定义中,按以下方式处理:

  • from OIPModule.py import OIPModule
  • 在__init__函数中,加入self.oip = OIPModule()
  • 在forward函数中,加入map = self.oip(map) 即可完成模块插入

数据集

数据集与下载链接:

SUN RGB-D

NYUv2

基础代码来自公开仓库

ESANet

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages