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生成的数据集:

  1. 可视化当前目录:
nvdu_viz
  1. 可视化相对路径:
nvdu_viz ../a_dataset
  1. 可视化绝对路径:
nvdu_viz ~/data/dataset
  1. 使用过滤器显示帧的不同方面:
nvdu_viz dataset_path --name_filters *.left.png *.right.png

使用不同的注释数据可视化一组图像:

  1. 相机和对象设置文件位于图像目录中:
nvdu_viz image_directory_here -a annotation_directory_here
  1. 相机和对象设置文件不在图像目录中:
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

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
多次调用预期方法的java捕获参数(EasyMock)   java我试图得到某个学生的平均值,但它将所有值相加   使用Java更新文件中所有出现的字符串   java从依赖包导入Spring配置(*.xml)   在Java中,如何从不同的偏移量读取大块的字符串文件?   java 安卓 studio未在windows 8中运行   java getResourceAsStream()不读取任何内容   java Google Collections 1.0是否已经专业化?   模型视图控制器ASP。NET母版页在Java中的等效性   计时器TimerTask类在java中只能运行一次   多级继承和foreach Java循环   excel csv到xsl java,有一列带有某种货币(如$400)   java在来电时多次更改BlackBerry的callIncoming()   java通过JNLP启动JavaFX2.0应用程序会引发异常