Enhancing Door Detection in Architectural Floorplans with Directional Analysis

rolurq: <p>Hello everyone, I am currently working on a challenging project involving the detection of doors in architectural floorplans. While I have successfully implemented a fine-tuned YOLO model for door detection from vectorized drawings in PDF floorplans, I am now faced with the task of determining the direction in which the detected doors open.</p>
<p>To provide a bit more context, my current solution involves leveraging a YOLO model trained specifically for door detection, and it has shown promising results in identifying the location of doors within floorplans. However, I am seeking assistance and insights on incorporating a solution that can accurately determine the direction in which each detected door opens, ideally in the form of a vector.</p>
<p>My first idea was to work with the data inside the PDF. Extracting the drawings in order to find the door arc, determine its center point and this, together with the arc start or end point should form the vector in which the door opens. However, the correct point from the arc may be either of the start or end, the only way of correctly determining it is to find the one that is closest to the door sheet, but in some floorplans, or even in different doors of the same one, the representation of a wall and a door sheet may look very similar and thus the correct point is hard to determine in this way.</p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/0/9/0976044909619f9fcbd0d0fc8238e226964a9756.png" data-download-href="https://discuss.pytorch.org/uploads/default/0976044909619f9fcbd0d0fc8238e226964a9756" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/0/9/0976044909619f9fcbd0d0fc8238e226964a9756_2_374x500.png" alt="image" data-base62-sha1="1lH96O3kgqiLXcHbThBv4Y3iQN8" width="374" height="500" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/0/9/0976044909619f9fcbd0d0fc8238e226964a9756_2_374x500.png, https://discuss.pytorch.org/uploads/default/optimized/3X/0/9/0976044909619f9fcbd0d0fc8238e226964a9756_2_561x750.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/0/9/0976044909619f9fcbd0d0fc8238e226964a9756.png 2x" data-dominant-color="FCF9F9"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">715×954 68.9 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p>My next idea, which I’m working on implementing at the moment, is to make the model learn the door direction alongside the detection. I’m thinking in incorporating a layer or having a separate model that can output the direction of the door using the detected area.</p>
<p>I would be immensely grateful for any insights, resources, or experiences you can share to help me overcome this challenge. If there are specific details or information you need to provide more targeted assistance, please feel free to ask.</p>