search

NVIDIA-AI-IOT / deepstream_tao_apps

Sample apps to demonstrate how to deploy models trained with TAO on DeepStream
MIT License
380 stars 97 forks source link

custom bounding box parsing function need more feilds #61

Open dongfangduoshou123 opened 2 years ago

dongfangduoshou123 commented 2 years ago

like the yolov4's detection custom parse function implemented in this repo, its output is not identity to the tao output. I know it is because the yolov4 parse need a ratio param, which is computed from frame_original_h, frame_original_w, network_input_w and network_input_w, but in current detection parse function interface, the frame_orignal_h and orignal_w can not be obtained yet.

So, I think the current detection parse function interface should fix this bug like to add feild like frame_orignal_h and frame_orignal_w in the NvDsInferNetworkInfo struct. current: typedef struct { / Holds the input width for the model. */ unsigned int width; /* Holds the input height for the model. / unsigned int height; / Holds the number of input channels for the model. */ unsigned int channels; } NvDsInferNetworkInfo; after added: typedef struct { / Holds the input width for the model. */ unsigned int width; /* Holds the input height for the model. / unsigned int height; / Holds the number of input channels for the model. */ unsigned int channels;

     unsigned int frame_orignal_h;
      unsigned int frame_orignal_w;
    } NvDsInferNetworkInfo;