dusty-nv / ros_deep_learning

Deep learning inference nodes for ROS / ROS2 with support for NVIDIA Jetson and TensorRT
887 stars 258 forks source link

class mask all 0s #64

Open RGring opened 3 years ago

RGring commented 3 years ago

Hi @dusty-nv, I this repo with a custom segmentation network. I spotted some copy-paste typos here: https://github.com/dusty-nv/ros_deep_learning/blob/master/src/node_segnet.cpp#L194 https://github.com/dusty-nv/ros_deep_learning/blob/master/src/node_segnet.cpp#L208

This fixed the correct output of the color mask and overlay. Though, the class mask is all 0. Do you have an idea why?