Bugs Fixed

requirements.txt

@@ -1,7 +1,5 @@
 sahi==0.11.14
 streamlit-image-comparison==0.0.4
-streamlit==1.22.0
+streamlit
 ultralyticsplus
-poppler-utils
-pdf2image
-PyPDF4
+

utils.py

@@ -1,8 +1,6 @@
 import numpy
 import sahi.predict
 import sahi.utils
-import PyPDF4
-from pdf2image import convert_from_path
 from PIL import Image
 
 TEMP_DIR = "temp"
@@ -15,55 +13,25 @@ def sahi_yolov8m_inference(
     slice_width=512,
     overlap_height_ratio=0.1,
     overlap_width_ratio=0.1,
-    image_size=640,
-    postprocess_match_threshold=0.5,
+    image_size=1024,
+    postprocess_match_threshold=0.75,
 ):
-
-    # standard inference
-    detection_model.image_size = image_size
-    prediction_result_1 = sahi.predict.get_prediction(
-        image=image, detection_model=detection_model
-    )
-    visual_result_1 = sahi.utils.cv.visualize_object_predictions(
-        image=numpy.array(image),
-        object_prediction_list=prediction_result_1.object_prediction_list,
-    )
-    output_1 = Image.fromarray(visual_result_1["image"])
-
     # sliced inference
-    prediction_result_2 = sahi.predict.get_sliced_prediction(
+    prediction_result = sahi.predict.get_sliced_prediction(
         image=image,
         detection_model=detection_model,
+        image_size = image_size,
         slice_height=slice_height,
         slice_width=slice_width,
         overlap_height_ratio=overlap_height_ratio,
         overlap_width_ratio=overlap_width_ratio,
         postprocess_match_threshold=postprocess_match_threshold,
     )
-    visual_result_2 = sahi.utils.cv.visualize_object_predictions(
+    visual_result = sahi.utils.cv.visualize_object_predictions(
         image=numpy.array(image),
-        object_prediction_list=prediction_result_2.object_prediction_list,
+        object_prediction_list=prediction_result.object_prediction_list,
     )
 
-    output_2 = Image.fromarray(visual_result_2["image"])
-
-    return output_1, output_2
+    output = Image.fromarray(visual_result["image"])
 
-# 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')
+    return output