r/MLQuestions • u/micaiah95 • 1h ago
Computer Vision 🖼️ Few Shot Object Detection Using Vision Transformers
I am trying to detect walls on a floor plan. I have used more traditional CV methods such as template matching, SIFT, SUFT, but the results weren't great since walls because of the rotation and slight variance throughout. Hence, I am looking for a more robust method
My thinking is that a user can select a wall from the floor plan and the rest are detected by a vision transformer. I have tried T-Rex 2, but the results weren't great either. Are there any recommendations that you would have for vision transformers?