File tree Expand file tree Collapse file tree 6 files changed +14
-9
lines changed
Expand file tree Collapse file tree 6 files changed +14
-9
lines changed Original file line number Diff line number Diff line change @@ -58,6 +58,10 @@ components:
5858 network_output_type : softmax
5959 data_format : NHWC
6060 cuda_stream_pool : stream
61+ - type : nvidia::gxf::MemoryAvailableSchedulingTerm
62+ parameters :
63+ allocator : allocator
64+ min_blocks : 1
6165---
6266name : segmentation_broadcaster
6367components :
Original file line number Diff line number Diff line change @@ -63,7 +63,7 @@ def generate_launch_description():
6363 description = 'A list of input tensor binding names (specified by model)' ),
6464 DeclareLaunchArgument (
6565 'input_tensor_formats' ,
66- default_value = '["nitros_tensor_list_nchw_rgb_f32 "]' ,
66+ default_value = '["nitros_tensor_list_nchw_bgr_f32 "]' ,
6767 description = 'The nitros format of the input tensors' ),
6868 DeclareLaunchArgument (
6969 'output_tensor_names' ,
@@ -75,7 +75,7 @@ def generate_launch_description():
7575 description = 'A list of output tensor binding names (specified by model)' ),
7676 DeclareLaunchArgument (
7777 'output_tensor_formats' ,
78- default_value = '["nitros_tensor_list_nhwc_rgb_f32 "]' ,
78+ default_value = '["nitros_tensor_list_nhwc_bgr_f32 "]' ,
7979 description = 'The nitros format of the output tensors' ),
8080 DeclareLaunchArgument (
8181 'network_output_type' ,
Original file line number Diff line number Diff line change @@ -67,7 +67,7 @@ def generate_launch_description():
6767 description = 'A list of input tensor binding names (specified by model)' ),
6868 DeclareLaunchArgument (
6969 'input_tensor_formats' ,
70- default_value = '["nitros_tensor_list_nchw_rgb_f32 "]' ,
70+ default_value = '["nitros_tensor_list_nchw_bgr_f32 "]' ,
7171 description = 'The nitros format of the input tensors' ),
7272 DeclareLaunchArgument (
7373 'output_tensor_names' ,
@@ -79,7 +79,7 @@ def generate_launch_description():
7979 description = 'A list of output tensor binding names (specified by model)' ),
8080 DeclareLaunchArgument (
8181 'output_tensor_formats' ,
82- default_value = '["nitros_tensor_list_nhwc_rgb_f32 "]' ,
82+ default_value = '["nitros_tensor_list_nhwc_bgr_f32 "]' ,
8383 description = 'The nitros format of the output tensors' ),
8484 DeclareLaunchArgument (
8585 'tensorrt_verbose' ,
@@ -145,6 +145,7 @@ def generate_launch_description():
145145 'network_image_height' : network_image_height ,
146146 'image_mean' : encoder_image_mean ,
147147 'image_stddev' : encoder_image_stddev ,
148+ 'enable_padding' : True ,
148149 }],
149150 remappings = [('encoded_tensor' , 'tensor_pub' )]
150151 )
Original file line number Diff line number Diff line change @@ -71,7 +71,7 @@ def generate_launch_description():
7171 description = 'A list of input tensor binding names (specified by model)' ),
7272 DeclareLaunchArgument (
7373 'input_tensor_formats' ,
74- default_value = '["nitros_tensor_list_nchw_rgb_f32 "]' ,
74+ default_value = '["nitros_tensor_list_nchw_bgr_f32 "]' ,
7575 description = 'The nitros format of the input tensors' ),
7676 DeclareLaunchArgument (
7777 'output_tensor_names' ,
@@ -83,7 +83,7 @@ def generate_launch_description():
8383 description = 'A list of output tensor binding names (specified by model)' ),
8484 DeclareLaunchArgument (
8585 'output_tensor_formats' ,
86- default_value = '["nitros_tensor_list_nhwc_rgb_f32 "]' ,
86+ default_value = '["nitros_tensor_list_nhwc_bgr_f32 "]' ,
8787 description = 'The nitros format of the output tensors' ),
8888 DeclareLaunchArgument (
8989 'network_output_type' ,
Original file line number Diff line number Diff line change @@ -71,7 +71,7 @@ def generate_launch_description():
7171 description = 'A list of input tensor binding names (specified by model)' ),
7272 DeclareLaunchArgument (
7373 'input_tensor_formats' ,
74- default_value = '["nitros_tensor_list_nchw_rgb_f32 "]' ,
74+ default_value = '["nitros_tensor_list_nchw_bgr_f32 "]' ,
7575 description = 'The nitros format of the input tensors' ),
7676 DeclareLaunchArgument (
7777 'output_tensor_names' ,
@@ -83,7 +83,7 @@ def generate_launch_description():
8383 description = 'A list of output tensor binding names (specified by model)' ),
8484 DeclareLaunchArgument (
8585 'output_tensor_formats' ,
86- default_value = '["nitros_tensor_list_nhwc_rgb_f32 "]' ,
86+ default_value = '["nitros_tensor_list_nhwc_bgr_f32 "]' ,
8787 description = 'The nitros format of the output tensors' ),
8888 DeclareLaunchArgument (
8989 'network_output_type' ,
Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ SPDX-License-Identifier: Apache-2.0
2121<?xml-model href =" http://download.ros.org/schema/package_format3.xsd" schematypens =" http://www.w3.org/2001/XMLSchema" ?>
2222<package format =" 3" >
2323 <name >isaac_ros_unet</name >
24- <version >2.0 .0</version >
24+ <version >2.1 .0</version >
2525 <description >U-Net model processing</description >
2626
2727 <
maintainer email =
" [email protected] " >Hemal Shah</
maintainer >
You can’t perform that action at this time.
0 commit comments