pytorch-image-models icon indicating copy to clipboard operation
pytorch-image-models copied to clipboard

[FEATURE]features_only option on vision transformer

Open 201820894 opened this issue 2 years ago • 0 comments

I'm trying to apply swin v2 as a backbone of dense prediction tasks such as depth estimation of sementic segmentation.

However, I found that features_only option is unavailable on vision transformer models.

Is there any other ways to extract features from each stages without using features_only argument on vision transformer?

201820894 avatar Dec 02 '22 08:12 201820894