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未从Facebook OAuth接收名字、姓氏和姓氏   我自己代码中的java ConcurrentModificationException   java在Android上获得单像素TIFF图像   java图形输出中的swing集成弹出窗口“通知”   将数据库表中的所有数据集插入jTable Java   java如何使用Hibernate获取不完整的集合?   打印无法从java连接到打印机   java使用分隔符拆分带引号的字符串   java Axis2禁用严格验证(wsdl2java中的Eosv)允许什么,如何使其更加严格?   java有人能帮我处理循环代码吗   java将JsonObject数组转换为整数   JavaXWiki扩展:检测事件空间创建   java如何设置图表的文本方向以使用poi旋转所有文本?   eche RecyclerView项的java Set自定义字体   java单元测试带有私有构造函数的spring组件,无需注入   用户界面如何在java中动态显示一组多个元素?   如何从java应用程序调用IDL(交互式数据语言)?   Java的内置库实现