search

HAL-42 / FMA-WSSS

This repo is a implementation of the Foundation Model Assisted Weakly Supervised Semantic Segmentation. The code is developed based on the Pytorch framework.
Apache License 2.0
41 stars 4 forks source link

SegVisualizationHook.__init__() got an unexpected keyword argument 'save_prob' #20

Open J-Lam0 opened 6 months ago

J-Lam0 commented 6 months ago

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'

HAL-42 commented 6 months ago

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.