CompletionFormer 点云补全 学习笔记

目录

依赖项

数据集

[KITTI DC](#KITTI DC)

[NYUv2 提供深度图](#NYUv2 提供深度图)

模型


依赖项

复制代码
conda create -n completionformer python=3.8
conda activate completionformer
# For CUDA Version == 11.3
pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113
pip install mmcv-full==1.4.4 mmsegmentation==0.22.1  
pip install timm tqdm thop tensorboardX opencv-python ipdb h5py ipython Pillow==9.5.0 

dcv v2

数据集

KITTI DC

depth velodyne 深度测速仪

NYUv2 提供深度图

复制代码
$ cd PATH_TO_DOWNLOAD
$ wget http://datasets.lids.mit.edu/sparse-to-dense/data/nyudepthv2.tar.gz
$ tar -xvf nyudepthv2.tar.gz

After that, you will get a data structure as follows:

复制代码
nyudepthv2
├── train
│    ├── basement_0001a
│    │    ├── 00001.h5
│    │    └── ...
│    ├── basement_0001b
│    │    ├── 00001.h5
│    │    └── ...
│    └── ...
└── val
    └── official
        ├── 00001.h5
        └── ...

Note that the original full NYUv2 dataset is available at the official website.

After preparing the dataset, you should generate a json file containing paths to individual images.

复制代码
$ cd THIS_PROJECT_ROOT/utils
$ python generate_json_NYUDepthV2.py --path_root PATH_TO_NYUv2

模型

318M

相关推荐
"菠萝"1 分钟前
C#知识学习-017(修饰符_6)
学习·c#
爱学习的uu7 分钟前
CURSOR最新使用指南及使用思路
人工智能·笔记·python·软件工程
YuCaiH13 分钟前
Linux文件处理
linux·笔记·嵌入式
早睡冠军候选人23 分钟前
Ansible学习----Ansible Playbook
运维·服务器·学习·云原生·容器·ansible
Cathy Bryant25 分钟前
大模型损失函数(二):KL散度(Kullback-Leibler divergence)
笔记·神经网络·机器学习·数学建模·transformer
楼田莉子30 分钟前
C++学习:C++11扩展:constexpr特性
开发语言·c++·学习
懒羊羊不懒@34 分钟前
Java基础语法—最小单位、及注释
java·c语言·开发语言·数据结构·学习·算法
qq_398586541 小时前
Threejs入门学习笔记
javascript·笔记·学习
hour_go1 小时前
TCP/IP协议相关知识点
网络·笔记·网络协议·tcp/ip
潘达斯奈基~2 小时前
在使用spark的applyInPandas方法过程中,遇到类型冲突问题如何解决
大数据·笔记