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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions