文件名称：matlab渐变颜色代码-MonoDepth-FPN-PyTorch:特征金字塔网络的单图像深度估计
文件大小：3.52MB
文件格式：ZIP
更新时间：2024-06-08 21:16:25
系统开源
matlab渐变颜色代码MonoDepth-FPN-PyTorch 一个简单的端到端模型，可在PyTorch中实现深度预测的最新性能。 我们使用功能金字塔网络（FPN）骨架来从单个输入RGB图像估计深度图。 我们在NYU深度V2数据集（官方拆分）和KITTI数据集（本征拆分）上测试了模型的性能。 要求 的Python 3 Jupyter Notebook（用于可视化） 火炬用PyTorch 0.3.0.post4测试 CUDA 8（如果使用CUDA） 跑步 python3 main_fpn.py --cuda --bs 6 要从保存的模型继续训练，请使用 python3 main_fpn.py --cuda --bs 6 --r True --checkepoch 10 要可视化重建的数据，请在vis.ipynb中运行jupyter笔记本。 数据处理 NYU深度V2数据集 包含各种室内场景，其中249个用于训练的场景和215个用于测试的场景。 我们使用官方分组进行了培训和测试。 这为数据处理提供了非常详细的演练和matlab代码。 在先前的工作之后，我们使用了官方工具箱，该工具箱使用了
【文件预览】：
MonoDepth-FPN-PyTorch-master
----.gitignore(25B)
----kitti_comparison.png(593KB)
----dataset()
--------.gitignore(23B)
--------augment.py(5KB)
--------.ipynb_checkpoints()
--------nyuv2_dataset.py(6KB)
--------__init__.py(0B)
--------dataloader.py(3KB)
--------kitti_dataset.py(2KB)
--------constants.py(27B)
----nyuv2_performance.png(97KB)
----architecture.png(202KB)
----main_fpn.py(17KB)
----LICENSE(1KB)
----comparison_nyuv2.png(1.13MB)
----README.md(10KB)
----kitti_performance.png(114KB)
----vis.ipynb(1.03MB)
----metrics.png(84KB)
----kitti.png(352KB)
----model_fpn.py(4KB)