nvidia数据集实用程序,
nvdu的Python项目详细描述
nvidia数据集实用程序(nvdu)
这个项目是一个python脚本的集合,用于帮助使用数据集进行深入学习。例如,可视化与nvidia深度学习数据集合成器(ndds)生成的捕获传感器图像相关联的注释数据https://github.com/NVIDIA/Dataset_Synthesizer。
This module depends on OpenCV-python which currently doesn't work with Python 3.7.
使用nvdu可视化的数据集框架示例,显示带注释对象的轴和三维长方体。
目录
安装
从PIP安装:
pip install nvdu
从源代码git repo安装:
克隆回购协议
使用ssh路径:
git clone ssh://git@github.com:12051/NVIDIA/Dataset_Utilities.git
使用https路径:
git clone https://github.com/NVIDIA/Dataset_Utilities.git
进入克隆的repo目录
cd Dataset_Utilities
本地安装
pip install -e .
nvdu_ycb
nvdu-ycb命令帮助下载、提取并对齐ycb三维模型(在fat数据集中使用:http://research.nvidia.com/publication/2018-06_Falling-Things)。
用法
usage: nvdu_ycb [-h] [-s] [-l] [ycb_object_name]
NVDU YCB models Support
positional arguments:
ycb_object_name Name of the YCB object to check.
optional arguments:
-h, --help show this help message and exit
-s, --setup Setup the YCB models for the FAT dataset
-l, --list List all the supported YCB objects
注意:如果在尝试使用nvdu-viz之前不运行nvdu_ycb --setup
,可视化工具将无法找到要覆盖的ycb对象的三维模型。
nvdu_即
nvdu-viz命令使用ndds格式可视化带注释的数据集。
用法
nvdu_viz [-h] [-a DATA_ANNOT_DIR] [-s SIZE SIZE]
[-o OBJECT_SETTINGS_PATH] [-c CAMERA_SETTINGS_PATH]
[-m MODEL_DIR] [-n [NAME_FILTERS [NAME_FILTERS ...]]]
[--fps FPS] [--auto_change] [-e EXPORT_DIR] [--auto_export]
[--ignore_fixed_transform]
[dataset_dir]
NVDU Data Visualiser
positional arguments:
dataset_dir Dataset directory. This is where all the images
(required) and annotation info (optional) are.
Defaults to the current directory.
optional arguments:
-h, --help show this help message and exit.
-a DATA_ANNOT_DIR, --data_annot_dir DATA_ANNOT_DIR
Directory path - where to find the annotation data.
Defaults to be the same directory as the dataset
directory.
-s SIZE SIZE, --size SIZE SIZE
Window's size: [width, height]. If not specified then
the window is sized to fit the resolution of the camera.
-o OBJECT_SETTINGS_PATH, --object_settings_path OBJECT_SETTINGS_PATH
Object settings file path.
-c CAMERA_SETTINGS_PATH, --camera_settings_path CAMERA_SETTINGS_PATH
Camera settings file path.
-n [NAME_FILTERS [NAME_FILTERS ...]], --name_filters [NAME_FILTERS [NAME_FILTERS ...]]
The name filter of each frame. e.g: *.png.
--fps FPS How fast to automatically change frame.
--auto_change When using this flag, the visualizer will automatically
change the frame.
-e EXPORT_DIR, --export_dir EXPORT_DIR
Directory path - where to store the visualized images.
If this is set, the script will automatically export
the visualized image to the export directory.
--auto_export When using this flag, the visualizer will automatically
export the visualized frame to an image file in the
`export_dir` directory.
--ignore_fixed_transform
When using this flag, the visualizer will not use the
fixed transform matrix for the 3d model.
注意:nvdu_viz
脚本可以在任何目录下工作
示例
可视化由ndds生成的数据集:
nvdu_viz
nvdu_viz ../a_dataset
nvdu_viz ~/data/dataset
nvdu_viz dataset_path --name_filters *.left.png *.right.png
使用不同的注释数据可视化一组图像:
nvdu_viz image_directory_here -a annotation_directory_here
nvdu_viz image_directory_here -a annotation_directory_here -c camera_setting_path_here -o object_setting_path_here
控件
可视化选项:
F3 - Toggle the 2d cuboid
F4 - Toggle the 3d cuboid
F5 - Toggle the 3d models
F6 - Toggle the axes
F7 - Toggle the overlay frame name
1 - Render the 3d models normally
2 - Render the 3d models using only the edge lines
3 - Render the 3d models as point clouds
其他:
ESC - Quit the visualizer
Right - Go to the next frame
Left - Go to the previous frame
Up - Go to the next 100 frame
Down - Go to the previous 100 frame
Space - Toggle frame auto-changing
F12 - Toggle exporting the visualized frame to file
推荐PyPI第三方库