pytorch-image-models
pytorch-image-models copied to clipboard
[FEATURE]features_only option on vision transformer
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?