qhfan / RMT

(CVPR2024)RMT: Retentive Networks Meet Vision Transformer
272 stars 18 forks source link

detect 里面train.py 无法直接使用,缺少引用 #29

Open 13579sjj opened 1 month ago

13579sjj commented 1 month ago

493dfdade51906a293bc5b8690be5b2 请问这个怎么解决

qhfan commented 1 month ago

cd detect bash install.sh

13579sjj commented 1 month ago

请问 我直接用train.sh 显示没有权限,按照网上的加权限也不行 我直接使用 python tools/train.py configs/RMT/maskrcnn_t_1x.py --cfg-options model.pretrained=/home/yry/sjj/RMT-main/RMT-T.pth model.backbone.use_checkpoint=True 这个代码会报KeyError: "MaskRCNN: 'RMT is not in the models registry'" 请问 在哪添加这个模型,谢谢!