Merge pull request #3 from NVIDIA-ISAAC-ROS/release-dp-1.1

Isaac ROS 0.11.0 (DP1.1)

Author: hemalshahNV

README.md

@@ -50,11 +50,11 @@ add_library(bi3d_postprocess SHARED IMPORTED)
 if(${ARCHITECTURE} STREQUAL "x86_64")
   set_property(
     TARGET bi3d_postprocess PROPERTY IMPORTED_LOCATION
-    ${CMAKE_CURRENT_SOURCE_DIR}/cvcore/cvcore_bi3d/lib/x86_64/libgxf_bi3d_postprocess.so)
+    ${CMAKE_CURRENT_SOURCE_DIR}/cvcore/cvcore_bi3d/lib/x86_64/libgxf_bi3d_postprocessor.so)
 elseif(${ARCHITECTURE} STREQUAL "aarch64")
   set_property(
     TARGET bi3d_postprocess PROPERTY IMPORTED_LOCATION
-    ${CMAKE_CURRENT_SOURCE_DIR}/cvcore/cvcore_bi3d/lib/jetpack50/libgxf_bi3d_postprocess.so)
+    ${CMAKE_CURRENT_SOURCE_DIR}/cvcore/cvcore_bi3d/lib/jetpack50/libgxf_bi3d_postprocessor.so)
 endif()
 
 # bi3d_node

isaac_ros_bi3d/CMakeLists.txt

@@ -6,13 +6,14 @@
 # and any modifications thereto.  Any use, reproduction, disclosure or
 # distribution of this software and related documentation without an express
 # license agreement from NVIDIA CORPORATION is strictly prohibited.
+##################################################
 ---
 name: sync
 components:
 - name: data_receiver_left
   type: nvidia::gxf::DoubleBufferReceiver
   parameters:
-    capacity: 1
+    capacity: 12
     policy: 0
 - type: nvidia::gxf::MessageAvailableSchedulingTerm
   parameters:
@@ -21,7 +22,7 @@ components:
 - name: data_receiver_right
   type: nvidia::gxf::DoubleBufferReceiver
   parameters:
-    capacity: 1
+    capacity: 12
     policy: 0
 - type: nvidia::gxf::MessageAvailableSchedulingTerm
   parameters:
@@ -100,17 +101,49 @@ components:
   parameters:
     capacity: 1
 ---
+name: disparity_roundrobin
+components:
+- name: data_receiver
+  type: nvidia::gxf::DoubleBufferReceiver
+  parameters:
+    capacity: 1
+    policy: 0
+- type: nvidia::gxf::MessageAvailableSchedulingTerm
+  parameters:
+    receiver: data_receiver
+    min_size: 1
+- type: nvidia::gxf::Broadcast
+  parameters:
+    source: data_receiver
+    mode: RoundRobin
+- name: data_transmitter_dla0
+  type: nvidia::gxf::DoubleBufferTransmitter
+  parameters:
+    capacity: 1
+- type: nvidia::gxf::DownstreamReceptiveSchedulingTerm
+  parameters:
+    transmitter: data_transmitter_dla0
+    min_size: 1
+- name: data_transmitter_dla1
+  type: nvidia::gxf::DoubleBufferTransmitter
+  parameters:
+    capacity: 1
+- type: nvidia::gxf::DownstreamReceptiveSchedulingTerm
+  parameters:
+    transmitter: data_transmitter_dla1
+    min_size: 1
+---
 name: gather
 components:
 - name: input_dla0
   type: nvidia::gxf::DoubleBufferReceiver
   parameters:
-    capacity: 12
+    capacity: 6
     policy: 0
 - name: input_dla1
   type: nvidia::gxf::DoubleBufferReceiver
   parameters:
-    capacity: 12
+    capacity: 6
     policy: 0
 - type: nvidia::gxf::MultiMessageAvailableSchedulingTerm
   parameters:
@@ -148,6 +181,15 @@ components:
   parameters:
     receiver: data_receiver_right
     min_size: 1
+- name: data_receiver_disparity
+  type: nvidia::gxf::DoubleBufferReceiver
+  parameters:
+    capacity: 6
+    policy: 0
+- type: nvidia::gxf::MessageAvailableSchedulingTerm
+  parameters:
+    receiver: data_receiver_disparity
+    min_size: 1
 - name: data_transmitter
   type: nvidia::gxf::DoubleBufferTransmitter
   parameters:
@@ -161,7 +203,13 @@ components:
   parameters:
     storage_type: 1
     block_size: 2211840
-    num_blocks: 50
+    num_blocks: 12
+- name: forward_pool
+  type: nvidia::gxf::BlockMemoryPool
+  parameters:
+    storage_type: 0
+    block_size: 256
+    num_blocks: 12
 - name: stream
   type: nvidia::gxf::CudaStreamPool
   parameters:
@@ -176,10 +224,12 @@ components:
     right_image_name: "right_image"
     output_name: "disparity"
     pool: pool
+    forward_pool: forward_pool
     stream_pool: stream
     left_image_receiver: data_receiver_left
     right_image_receiver: data_receiver_right
     output_transmitter: data_transmitter
+    disparity_receiver: data_receiver_disparity
 
     image_type: "RGB_U8"
     pixel_mean: [0.0078431372549, 0.0078431372549, 0.0078431372549]
@@ -206,7 +256,6 @@ components:
     threshold_value_low: 0.0
     threshold_value_high: 1.0
     max_disparity_levels: 64
-    disparity_values: [18]
 ---
 name: bi3d_dla1
 components:
@@ -228,6 +277,15 @@ components:
   parameters:
     receiver: data_receiver_right
     min_size: 1
+- name: data_receiver_disparity
+  type: nvidia::gxf::DoubleBufferReceiver
+  parameters:
+    capacity: 6
+    policy: 0
+- type: nvidia::gxf::MessageAvailableSchedulingTerm
+  parameters:
+    receiver: data_receiver_disparity
+    min_size: 1
 - name: data_transmitter
   type: nvidia::gxf::DoubleBufferTransmitter
   parameters:
@@ -241,7 +299,13 @@ components:
   parameters:
     storage_type: 1
     block_size: 2211840 
-    num_blocks: 50
+    num_blocks: 12
+- name: forward_pool
+  type: nvidia::gxf::BlockMemoryPool
+  parameters:
+    storage_type: 0
+    block_size: 256
+    num_blocks: 12
 - name: stream
   type: nvidia::gxf::CudaStreamPool
   parameters:
@@ -256,10 +320,12 @@ components:
     right_image_name: "right_image"
     output_name: "disparity"
     pool: pool
+    forward_pool: forward_pool
     stream_pool: stream
     left_image_receiver: data_receiver_left
     right_image_receiver: data_receiver_right
     output_transmitter: data_transmitter
+    disparity_receiver: data_receiver_disparity
 
     image_type: "RGB_U8"
     pixel_mean: [0.0078431372549, 0.0078431372549, 0.0078431372549]
@@ -286,7 +352,6 @@ components:
     threshold_value_low: 0.0
     threshold_value_high: 1.0
     max_disparity_levels: 64
-    disparity_values: [18]
 ---
 name: sync_bi3d_dla0
 components:
@@ -343,7 +408,7 @@ components:
   parameters:
     storage_type: 1
     block_size: 2211840
-    num_blocks: 50
+    num_blocks: 24
 - name: data_receiver
   type: nvidia::gxf::DoubleBufferReceiver
   parameters:
@@ -361,12 +426,46 @@ components:
   parameters:
     transmitter: data_transmitter
     min_size: 1
-- type: nvidia::isaac_ros::Bi3DPostprocess
+- type: nvidia::isaac_ros::Bi3DPostprocessor
   parameters:
     bi3d_receiver: data_receiver
     output_transmitter: data_transmitter
     pool: pool
-    disparity_values: [18]
+    disparity_tensor_name: "disparity"
+    disparity_values_tensor_name: "bi3d_inference_disparities"
+---
+name: bi3d_output_message_splitter
+components:
+- name: data_receiver
+  type: nvidia::gxf::DoubleBufferReceiver
+  parameters:
+    capacity: 12
+    policy: 0
+- type: nvidia::gxf::MessageAvailableSchedulingTerm
+  parameters:
+    receiver: data_receiver
+    min_size: 1
+- name: disparity_image_transmitter
+  type: nvidia::gxf::DoubleBufferTransmitter
+  parameters:
+    capacity: 12
+- type: nvidia::gxf::DownstreamReceptiveSchedulingTerm
+  parameters:
+    transmitter: disparity_image_transmitter
+    min_size: 1
+- name: disparity_values_transmitter
+  type: nvidia::gxf::DoubleBufferTransmitter
+  parameters:
+    capacity: 12
+- type: nvidia::gxf::DownstreamReceptiveSchedulingTerm
+  parameters:
+    transmitter: disparity_values_transmitter
+    min_size: 1
+- type: nvidia::isaac_ros::Bi3DMessageSplitter
+  parameters:
+    receiver: data_receiver
+    disparity_image_transmitter: disparity_image_transmitter
+    disparity_values_transmitter: disparity_values_transmitter
 ---
 name: bi3d_output_vault
 components:
@@ -386,6 +485,24 @@ components:
     max_waiting_count: 1
     drop_waiting: false
 ---
+name: disparity_values_output_vault
+components:
+- name: input
+  type: nvidia::gxf::DoubleBufferReceiver
+  parameters:
+    capacity: 1
+    policy: 0
+- type: nvidia::gxf::MessageAvailableSchedulingTerm
+  parameters:
+    receiver: input
+    min_size: 1
+- name: vault
+  type: nvidia::gxf::Vault
+  parameters:
+    source: input
+    max_waiting_count: 1
+    drop_waiting: false
+---
 components:
 - name: connection
   type: nvidia::gxf::Connection
@@ -415,6 +532,20 @@ components:
     target: bi3d_dla0/data_receiver_right
 ---
 components:
+- name: connection
+  type: nvidia::gxf::Connection
+  parameters:
+    source: disparity_roundrobin/data_transmitter_dla0
+    target: bi3d_dla0/data_receiver_disparity
+---
+components:
+- name: connection
+  type: nvidia::gxf::Connection
+  parameters:
+    source: disparity_roundrobin/data_transmitter_dla1
+    target: bi3d_dla1/data_receiver_disparity
+---
+components:
 - name: connection
   type: nvidia::gxf::Connection
   parameters:
@@ -468,8 +599,22 @@ components:
   type: nvidia::gxf::Connection
   parameters:
     source: bi3d_postprocess/data_transmitter
+    target: bi3d_output_message_splitter/data_receiver
+---
+components:
+- name: connection
+  type: nvidia::gxf::Connection
+  parameters:
+    source: bi3d_output_message_splitter/disparity_image_transmitter
     target: bi3d_output_vault/input
 ---
+components:
+- name: connection
+  type: nvidia::gxf::Connection
+  parameters:
+    source: bi3d_output_message_splitter/disparity_values_transmitter
+    target: disparity_values_output_vault/input
+---
 name: scheduler
 components:
 - name: clock
@@ -480,5 +625,3 @@ components:
     worker_thread_number: 8
     stop_on_deadlock: false
 
-
-

isaac_ros_bi3d/config/bi3d_node.yaml

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c1c0238952b11a230afd491b9dfef8882522ab2623e490fa2cb57a66a643b849
+size 220688

isaac_ros_bi3d/cvcore/cvcore_bi3d/lib/jetpack50/libgxf_bi3d_postprocess.so

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5f0e347f150ce2d50ea66e3a87f2f509c65f321b23e83a0527c9719ffce5aa03
-size 2141032
+oid sha256:066f2da3f369df89e1a4954a549de96d2bce9ef5649cd34ed06e773582c2dd56
+size 2150416

isaac_ros_bi3d/cvcore/cvcore_bi3d/lib/jetpack50/libgxf_bi3d_postprocessor.so

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c26b35a91f9ce9b8aa416d33ce57227c59dfa876d89f85248b19006d9cc4303e
-size 295824
+oid sha256:a94e2f10048dfefdfec3c084b42fb42c17e7c6eaa7dd5078d5904fa6c1511888
+size 295992

isaac_ros_bi3d/cvcore/cvcore_bi3d/lib/jetpack50/libgxf_cvcore_bi3d.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c035c6b4e4f90c3883d3fb897828d3a13a64db71b61b10d5e08b0864c50796a2
+size 234576

isaac_ros_bi3d/cvcore/cvcore_bi3d/lib/jetpack50/libnvcv_tensorops.so

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ae494685b9c6bf8bcf09258426f65ff65c628eee20a618d7c000dd699d80719
-size 2288168
+oid sha256:da4da8b56d785e4d18ebfb7535186263a21d340a8716aca8681c7f4d1e914af4
+size 2297368