open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
8.04k stars 2.58k forks source link

Training of PP-MobileSeg #3344

Open rpl-alan-shangguan opened 1 year ago

rpl-alan-shangguan commented 1 year ago

Can I use mmsegmentation to train my own PP-MobileSeg model?

I didn't find loss_decode in pp_mobileseg config file, and When I add it to the decode_head and run the training, I got Traceback (most recent call last): File "tools/train.py", line 104, in main() File "tools/train.py", line 93, in main runner = Runner.from_cfg(cfg) File "/home/alan/Downloads/mmengine/mmengine/runner/runner.py", line 446, in from_cfg runner = cls( File "/home/alan/Downloads/mmengine/mmengine/runner/runner.py", line 413, in init self.model = self.build_model(model) File "/home/alan/Downloads/mmengine/mmengine/runner/runner.py", line 820, in build_model model = MODELS.build(model) File "/home/alan/Downloads/mmengine/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, kwargs, registry=self) File "/home/alan/Downloads/mmengine/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "/home/alan/Downloads/mmengine/mmengine/registry/build_functions.py", line 121, in build_from_cfg obj = obj_cls(args) # type: ignore File "/home/alan/Downloads/mmsegmentation/mmseg/models/segmentors/encoder_decoder.py", line 92, in init self._init_decode_head(decode_head) File "/home/alan/Downloads/mmsegmentation/mmseg/models/segmentors/encoder_decoder.py", line 102, in _init_decode_head self.decode_head = MODELS.build(decode_head) File "/home/alan/Downloads/mmengine/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, kwargs, registry=self) File "/home/alan/Downloads/mmengine/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "/home/alan/Downloads/mmengine/mmengine/registry/build_functions.py", line 121, in build_from_cfg obj = obj_cls(args) # type: ignore TypeError: init() got an unexpected keyword argument 'loss_decode'

JH95-ai commented 8 months ago

I also find this error .Have you solved it yet? @rpl-alan-shangguan

Qiang-Z commented 7 months ago

+1