
NIID-Net: NIID-Net的代码实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本项目实现了针对特定任务优化的神经网络模型NIID-Net,旨在解决非独立同分布数据环境下的学习问题。代码开源,便于研究与应用。
NIID-Net:适应室内场景内在图像分解的表面法线知识
依赖项:
Python 3.5
PyTorch 0.3.1(如果您使用 PyTorch > 0.3.1,请阅读相关文档)
TorchVision 0.2.1
对于其他依赖项,我们提供了一个 requirements.txt 文件。
数据集:
内在图像数据集
按照指示下载 CGI、IIW 和 SAW 数据集。请注意,Z. Li 和 N. Snavely 扩充了原始的和数据集。
如果您不打算训练模型,则可能不需要下载 CGI 数据集。
将这些数据集放置在项目目录下的 dataset 文件夹中。
最终的目录结构如下:
NIID-Net 项目
|---README.md
|---...
|---dataset
|---CGIntrinsics
|---intrinsics_final
| |---images
全部评论 (0)
还没有任何评论哟~


