Add focalnet backbone (#23104)

Adds FocalNet backbone to return features from all stages
2023-05-03 19:32:42 +03:00
parent ca7eb27ed5
commit 441658dd6c
10 changed files with 210 additions and 11 deletions
--- a/tests/test_backbone_common.py
+++ b/tests/test_backbone_common.py
@@ -135,6 +135,8 @@ class BackboneTesterMixin:
            # Verify num_features has been initialized in the backbone init
            self.assertIsNotNone(backbone.num_features)
            self.assertTrue(len(backbone.channels) == len(backbone.out_indices))
+            print(backbone.stage_names)
+            print(backbone.num_features)
            self.assertTrue(len(backbone.stage_names) == len(backbone.num_features))
            self.assertTrue(len(backbone.channels) <= len(backbone.num_features))
            self.assertTrue(len(backbone.out_feature_channels) == len(backbone.stage_names))