SegVisualizationHook.init() got an unexpected keyword argument 'save_prob'

Question

SegVisualizationHook.init() got an unexpected keyword argument 'save_prob'

Opened this issue a year ago · 1 comments

when I focus on stage 2, I use mmcv2.1.0 mmdet 3.3.0 mmengine 0.7.3 mmsegmentation 1.2.2 to train the segmentation network successfully. do u know why the error shown below occur when i test the segmentation network and how to solve it?
TypeError: class SegVisualizationHook in mmseg/engine/hooks/visualization_hook.py: SegVisualizationHook.init() got an unexpected keyword argument 'save_prob'

Answer 1 · 2024-04-29T05:22:19.000Z

The mmseg I use has been modified to support saving the logits generated during inference. Please use the mmseg provided by this repository. Alternatively, you can remove all save_prob items from the configuration file.