Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -271,6 +271,11 @@ def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpo
 
     #Annotation for drawin lines as in the markups
     for i in range(len(lines)):
+        l_points = [fitz.Point(*sanda[i][0])* derotationMatrix, fitz.Point(*lines[i][0])* derotationMatrix, fitz.Point(*lines[i][1])* derotationMatrix, fitz.Point(*sanda[i][1])* derotationMatrix]
+        annot = page.add_polyline_annot(l_points) 
+        annot.set_border(width=2, dashes=None)  # Optional border styling
+        annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
+        annot.update()
         # annot = page.add_line_annot(lines[i][0], lines[i][1])
         # # annot = page.add_line_annot(sanda[i][0], lines[i][0])
         # # annot = page.add_line_annot(sanda[i][1], lines[i][1])
@@ -278,23 +283,23 @@ def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpo
         # annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
         # annot.update()
        
-      annot = page.add_line_annot(fitz.Point(*lines[i][0]) * derotationMatrix, fitz.Point(*lines[i][1]) * derotationMatrix)
-      annot.set_border(width=2, dashes=None)  # Optional border styling
-      annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-      annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
-      annot.update()
-
-      annot = page.add_line_annot(fitz.Point(*sanda[i][0]) * derotationMatrix, fitz.Point(*lines[i][0]) * derotationMatrix)
-      annot.set_border(width=2, dashes=None)  # Optional border styling
-      annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-      annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
-      annot.update()
-
-      annot = page.add_line_annot(fitz.Point(*sanda[i][1]) * derotationMatrix, fitz.Point(*lines[i][1]) * derotationMatrix)
-      annot.set_border(width=2, dashes=None)  # Optional border styling
-      annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
-      annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
-      annot.update()
+      #annot = page.add_line_annot(fitz.Point(*lines[i][0]) * derotationMatrix, fitz.Point(*lines[i][1]) * derotationMatrix)
+      #annot.set_border(width=2, dashes=None)  # Optional border styling
+      #annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
+      #annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
+      #annot.update()
+
+      #annot = page.add_line_annot(fitz.Point(*sanda[i][0]) * derotationMatrix, fitz.Point(*lines[i][0]) * derotationMatrix)
+      #annot.set_border(width=2, dashes=None)  # Optional border styling
+      #annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
+      #annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
+      #annot.update()
+
+      #annot = page.add_line_annot(fitz.Point(*sanda[i][1]) * derotationMatrix, fitz.Point(*lines[i][1]) * derotationMatrix)
+      #annot.set_border(width=2, dashes=None)  # Optional border styling
+      #annot.set_colors(stroke=(1, 0, 0))  # Set the line color to red
+      #annot.set_info(content=str(1.0)+' m',subject='Perimeter Measurement', title="ADR Team")
+      #annot.update()
     # output_pdf_path = pdf_name+"_annotated.pdf"
     # Save the PDF with annotations