文件名称：数据融合matlab代码-DenseFusion:密集融合
文件大小：6.24MB
文件格式：ZIP
更新时间：2024-06-11 06:46:59
系统开源
数据融合matlab代码密集融合闪电 表中的内容 概述 此存储库扩展了（。）的原始实现。我们重新实现了网络的某些部分，以允许进行以下扩展： 完整的Pytorch闪电整合 提供易于使用的Conda环境 PyTorch评估管道 在多个GPU上进行培训 批量培训＃ 将培训时间从多天减少到8小时 允许更多数据扩充 包括高级学习率安排 通常，这是Wang等人的论文“ DenseFusion：通过迭代密集融合进行6D对象姿势估计”的实现。 在和。 要求 CUDA 7.5 / 8.0 / 9.0（必需。由于对称对象的损耗计算（逐像素最近邻损耗），因此仅CPU会导致极慢的训练速度。） 数据集 这项工作在两个6D对象姿态估计数据集上进行了测试： ：培训和测试集如下。 训练集包括80个训练视频0000-0047和0060-0091（在我们的训练中以7帧作为间隙选择）和综合数据000000-079999。 测试集包括来自10个测试视频0048-0059的2949个关键帧。 ：下载（包括经过培训的经过评估的香草SegNet输出的测试结果）。 下载YCB_Video数据集，预处理的LineMOD数据集和训练有素
【文件预览】：
DenseFusion-Pytorch-1.0
----trained_models()
--------linemod()
--------ycb()
----src()
--------loss()
--------dataset()
--------lightning.py(28KB)
--------rotations()
--------helper()
--------eval()
--------visu()
----LICENSE(1KB)
----cfg()
--------fonts()
--------dataset()
--------exp()
--------env()
----README.md(10KB)
----lib()
--------transformations.py(65KB)
--------network.py(7KB)
--------extractors.py(4KB)
--------pspnet.py(2KB)
--------__pycache__()
----.vscode()
--------settings.json(156B)
----vanilla_segmentation()
--------trained_models()
--------segnet.py(6KB)
--------data_controller.py(4KB)
--------loss.py(721B)
--------logs()
--------train.py(5KB)
----download.sh(2KB)