-
Notifications
You must be signed in to change notification settings - Fork 42
Description
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 ?