Skip to content

Unable to use the pre-trained "ViTAE-B + RVSA ◊" with "tools\test.py" #55

@PoissonChasseur

Description

@PoissonChasseur

I am trying to simply use the pre-trained "ViTAE-B + RVSA ◊" on iSAID with the file "tools\test.py" , but I am getting the following error:

Traceback (most recent call last):
  File "tools/test.py", line 303, in <module>
    main()
  File "tools/test.py", line 203, in main
    model = build_segmentor(cfg.model, test_cfg=cfg.get('test_cfg'))
  File "C:\Users\vincentrm\D\other_features\.venv_RSP\lib\site-packages\mmseg\models\builder.py", line 58, in build_segmentor
v\utils\registry.py", line 60, in build_from_cfg
    raise type(e)(f'{obj_cls.__name__}: {e}')
KeyError: "EncoderDecoder: 'ViTAE_NC_Win_RVSA_V3_KVDIFF_WSZ7 is not in the models registry'"

I try to update the Tree-Directory for the the Github "RSP-main" with the files from this Github ("Remote-Sensing-RVSA"). However, even after I update the "init.py" file and put all the files from "Remote-Sensing-RVSA-main\Semantic Segmentation\mmseg\models\backbones" toward "RSP-main\Semantic Segmentation\mmseg\models\backbones", I am still getting this issue.

I am currently trying to run the file "tools\test.py" from the "RSP-main" folder. However, maybe it's not the right way to do this.

How to be able to make this code working, in order to simply run the pre-trained "ViTAE-B + RVSA ◊" (on iSAID) over some images ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions