【文件属性】:
文件名称:3D-to-2D-Distallation-for-Indoor-Scene-Parsing
文件大小:298KB
文件格式:ZIP
更新时间:2021-03-17 01:21:15
Python
用于室内场景解析的3D到2D蒸馏
介绍
此存储库是CVPR 2021(口服)中用于室内场景解析的3D到2D蒸馏的实现。 该代码基于 。
用法
要求:
硬件:4个GPU(更好的是> = 11G GPU内存)
软体:PyTorch> = 1.1.0,Python3, ,
克隆存储库:
git clone https://github.com/liuzhengzhe/3D-to-2D-Distallation-for-Indoor-Scene-Parsing
火车:
从下载提取的ScanNet-v2 2D数据和3D特征(提取代码:60oc),包括名为“ eachfeature”和“ featidx2”的文件夹。 然后将它们放在数据文件夹中:
忽略ScanNet中20个未使用的类别并生成新的实际情况
python data_process1.py
python data_proce
【文件预览】:
3D-to-2D-Distallation-for-Indoor-Scene-Parsing-master
----.gitignore(216B)
----data()
--------proj3dto2d_2.py(1KB)
--------meanvar()
--------proj3dto2d_1.py(3KB)
----README.md(3KB)
----3d-2d-distillation()
--------gene_list.py(141B)
--------list_val(304KB)
--------model()
--------dataset()
--------exp()
--------list(2.51MB)
--------util()
--------data_process2.py(956B)
--------config()
--------apex()
--------lib()
--------tool()
--------listtest(248KB)
--------data_process1.py(372B)