Merge branch 'FANPR-2'

arvindrajan92 · Mar 25, 2024 · 9b843b0 · 9b843b0
2 parents ad5ce35 + 621cc68
commit 9b843b0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fastanpr/detection.py b/fastanpr/detection.py
@@ -28,6 +28,6 @@ def run(self, images: List[np.ndarray]) -> List[List[Detection]]:
                 det_confs = detection.boxes.cpu().conf.numpy().tolist()
                 for det_box, det_conf in zip(det_boxes, det_confs):
                     x_min, x_max, y_min, y_max = det_box[0], det_box[2], det_box[1], det_box[3]
-                    image_detections.append(Detection(image[y_min:y_max+1, x_min:x_max+1, :], det_box[:4], det_conf))
+                    image_detections.append(Detection(image[y_min:y_max, x_min:x_max, :], det_box[:4], det_conf))
             results.append(image_detections)
         return results
diff --git a/setup.py b/setup.py
@@ -48,7 +48,7 @@
     packages=find_packages(),
     package_data={'': ['*.pt'], 'fastanpr': ['*.pt']},
     include_package_data=True,
-    install_requires=['ultralytics>=8.1.26', 'paddlepaddle>=2.6.0', 'paddleocr>=2.7.0.3'],
+    install_requires=['ultralytics>=8.1.34', 'PyMuPDF==1.24.0', 'paddlepaddle>=2.6.1', 'paddleocr>=2.7.2'],
     python_requires=PYTHON_REQUIRES,
     extras_require={
         'dev': ['pytest', 'pytest-asyncio', 'twine', 'python-Levenshtein', 'setuptools', 'wheel', 'twine', 'flake8']