Update files

app.py

@@ -100,7 +100,7 @@ with col2:
     st.markdown('##### Preview')
     # visualize input image
     if uploaded_file is not None:
-        image_file = convert_pdf_file(path=path)
+        #image_file = convert_pdf_file(path=path)
         image = Image.open(image_file)
     else:
         image = sahi.utils.cv.read_image_as_pil(IMAGE_TO_URL[radio])
@@ -163,7 +163,7 @@ with col2:
             img2=st.session_state["output_2"],
             label1='Uploaded Diagram',
             label2='Model Inference',
-            width=750,
+            width=1280,
             starting_position=50,
             show_labels=True,
             make_responsive=True,

utils.py

@@ -16,10 +16,7 @@ def sahi_yolov8m_inference(
     overlap_height_ratio=0.2,
     overlap_width_ratio=0.2,
     image_size=640,
-    #postprocess_type="GREEDYNMM",
-    #postprocess_match_metric="IOS",
     postprocess_match_threshold=0.5,
-    #postprocess_class_agnostic=False,
 ):
 
     # standard inference
@@ -41,10 +38,7 @@ def sahi_yolov8m_inference(
         slice_width=slice_width,
         overlap_height_ratio=overlap_height_ratio,
         overlap_width_ratio=overlap_width_ratio,
-        #postprocess_type=postprocess_type,
-        #postprocess_match_metric=postprocess_match_metric,
         postprocess_match_threshold=postprocess_match_threshold,
-        #postprocess_class_agnostic=postprocess_class_agnostic,
     )
     visual_result_2 = sahi.utils.cv.visualize_object_predictions(
         image=numpy.array(image),
@@ -55,21 +49,21 @@ def sahi_yolov8m_inference(
 
     return output_1, output_2
 
-def convert_pdf_file(
-    path,
-    #filename=name,
-    dpi=300,
-    image_width=4961,
-    image_heigth=3508,
-    grayscale=True,
-):
-    with open(path, 'rb') as pdf_file:
-        pdf_reader = PyPDF4.PdfFileReader(pdf_file, strict=False)
-        first_page = pdf_reader.getPage(0)
-        page_size = (first_page.mediaBox.getWidth(), first_page.mediaBox.getHeight())
-    if page_size[0] > page_size[1]:
-        image = convert_from_path(path, dpi=dpi, size=(image_width,image_heigth), grayscale=grayscale)
-    else:
-        image = convert_from_path(path, dpi=dpi, size=(image_heigth,image_width), grayscale=grayscale)
-    return image
-    #image[0].save(f'{path}/{filename}.png', 'PNG')
+# def convert_pdf_file(
+#     path,
+#     #filename=name,
+#     dpi=300,
+#     image_width=4961,
+#     image_heigth=3508,
+#     grayscale=True,
+# ):
+#     with open(path, 'rb') as pdf_file:
+#         pdf_reader = PyPDF4.PdfFileReader(pdf_file, strict=False)
+#         first_page = pdf_reader.getPage(0)
+#         page_size = (first_page.mediaBox.getWidth(), first_page.mediaBox.getHeight())
+#     if page_size[0] > page_size[1]:
+#         image = convert_from_path(path, dpi=dpi, size=(image_width,image_heigth), grayscale=grayscale)
+#     else:
+#         image = convert_from_path(path, dpi=dpi, size=(image_heigth,image_width), grayscale=grayscale)
+#     return image
+#     image[0].save(f'{path}/{filename}.png', 'PNG')