KeyError: 'OrientedRCNN is not in the models registry'

ViTAE-Transformer / Remote-Sensing-RVSA

The official repo for [TGRS'22] "Advancing Plain Vision Transformer Towards Remote Sensing Foundation Model"

MIT License

407 stars 31 forks source link

KeyError: 'OrientedRCNN is not in the models registry' #20

Open AbhaSinghPathania opened 1 year ago

AbhaSinghPathania commented 1 year ago

I have run the inference command using the following command: python image_demo.py demo/demo.jpg \ configs/obb/oriented_rcnn/vit_base_win/faster_rcnn_orpn_our_rsp_vitae-nc-base-win-rvsa_v3_wsz7_fpn_1x_dota10_ms_lr1e-4 _ldr75_dpr15.py \ checkpoints/vitae_rvsa_new.pth \ --device cpu

Its using image_demo.py file from OBBDetection. I am using MMCV version 1.6.2 and python version 3.9. How to reslove this error?

DotWang commented 1 year ago

@AbhaSinghPathania Do you install the OBBdetection?

AbhaSinghPathania commented 1 year ago

Yes. But it is not being used here in this code. For inference code for detection CLI provided in the repo where do we have to specify the path of images for inference. Is there something I am missing. I want to use RVSA on one of DOTA like image and want to run the inference code for the same. Can someone please specify how to run the inference for detection.

Justth1 commented 10 months ago

hey guys，do you solve this problem？I have the same problem