[FEA]: Remove hard-coded fields from GenerateVizFramesStage
#2136
Labels
feature request
New feature or request
GenerateVizFramesStage
#2136
Is this a new feature, an improvement, or a change to existing functionality?
Improvement
How would you describe the priority of this feature request
Medium
Please provide a clear description of problem this feature solves
The
GenerateVizFramesStage
(python/morpheus/morpheus/stages/postprocess/generate_viz_frames_stage.py
) contains hard-coded fields specific to the SID pipeline ex:Describe your ideal solution
The
idx2label
values should be fetched from theConfig.class_labels
attribute, and the DF field names could be constructor arguments.Alternately this class could be moved to
examples/sid_visualization
.Additional context
No response
Code of Conduct
The text was updated successfully, but these errors were encountered: