search

SwinTransformer / Swin-Transformer-Object-Detection

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.
https://arxiv.org/abs/2103.14030
Apache License 2.0
1.79k stars 379 forks source link

How to extract roi features of Swin #49

Open dong03 opened 3 years ago

dong03 commented 3 years ago

I'm trying to extract feature for each bbox via roi_extractor in mmdet. Currently, I run

from mmdet.models import build_roi_extractor
config_file = '../configs/swin/cascade_mask_rcnn_swin_base_patch4_window7_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py'
config = mmcv.Config.fromfile(config_file )
model = build_roi_extractor(config.model)

and get KeyError: 'CascadeRCNN is not in the roi_extractor registry'

Note that I've already run python setup.py develop.

Is there any other way or something i missed? Thanks

xhyandwyy commented 3 years ago

+1

Pter61 commented 2 years ago

+1