Update the main code to have the detection results in data as well as img type

src/deep_barcode_reader/__init__.py

@@ -1,3 +1,14 @@
-""" It can read different types of barcodes """
-
-__version__ = "0.2.0"
+""" It can read different types of barcodes """
+
+__version__ = "0.2.0"
+
+from .barcode import Wrapper, QRreader, BarcodeOpencv, BarcodeQRZbar
+from .base import ReaderResults
+
+__all__ = [
+    "Wrapper",
+    "QRreader",
+    "BarcodeOpencv",
+    "BarcodeQRZbar",
+    "ReaderResults",
+]

src/deep_barcode_reader/barcode.py

@@ -2,7 +2,7 @@
 
 # pylint: disable=E1101
 import logging
-from typing import Any
+from typing import Any, Tuple
 from dataclasses import dataclass, field
 from pathlib import Path
 
@@ -145,7 +145,7 @@ class Wrapper:
     method: str = field(default="opencv")
     model_size: str = field(default="l")
 
-    async def method_selection(self, data_path: str, result_path: str) -> Any:
+    async def method_selection(self, image: Any, result_path: str) -> Tuple[Any, Any]:
         """Wrap the method selection for barcode reader"""
         if self.method == "opencv":
             barcode: Any = BarcodeOpencv()
@@ -155,5 +155,6 @@ class Wrapper:
             barcode = QRreader(model_size=self.model_size)
         else:
             barcode = BarcodeOpencv()
-        detections = await barcode.detect_decode(cv2.imread(data_path))
-        return await detections.visualize_results_async(result_path)
+        detections = await barcode.detect_decode(image)
+        result_image = await detections.visualize_results_async(result_path)
+        return detections, result_image

src/deep_barcode_reader/logging.py

@@ -1,30 +1,30 @@
-"""Logger initialization"""
-
-import logging
-import logging.config
-from typing import Any, Optional
-
-
-def config_logger(loglevel: int) -> Any:
-    """Initialize a custom logger"""
-    default_logging_config = {
-        "version": 1,
-        "disable_existing_loggers": False,
-        "formatters": {
-            'standard': {
-                'format': '%(asctime)s - [%(levelname)s] [%(name)s.%(funcName)s:%(lineno)d (%(process)d)] | %(message)s',
-                'datefmt': '%Y-%m-%d %H:%M:%S',
-            },
-        },
-        "handlers": {
-            "console": {
-                "class": "logging.StreamHandler",
-                "formatter": "standard",
-            },
-        },
-        "root": {
-            "handlers": ["console"],
-            "level": loglevel,
-        },
-    }
-    logging.config.dictConfig(default_logging_config)
+"""Logger initialization"""
+
+import logging
+import logging.config
+from typing import Any
+
+
+def config_logger(loglevel: int) -> Any:
+    """Initialize a custom logger"""
+    default_logging_config = {
+        "version": 1,
+        "disable_existing_loggers": False,
+        "formatters": {
+            "standard": {
+                "format": "%(asctime)s - [%(levelname)s] [%(name)s.%(funcName)s:%(lineno)d (%(process)d)] | %(message)s",
+                "datefmt": "%Y-%m-%d %H:%M:%S",
+            },
+        },
+        "handlers": {
+            "console": {
+                "class": "logging.StreamHandler",
+                "formatter": "standard",
+            },
+        },
+        "root": {
+            "handlers": ["console"],
+            "level": loglevel,
+        },
+    }
+    logging.config.dictConfig(default_logging_config)

src/deep_barcode_reader/main.py

@@ -1,9 +1,11 @@
 """Run the main code for Deep-Barcode-Reader"""
 
+# pylint: disable=E1101
 from pathlib import Path
 import logging
 import asyncio
 import click
+import cv2
 
 
 from deep_barcode_reader import __version__
@@ -64,5 +66,7 @@ def deep_barcode_reader_cli(
 
     reader = Wrapper(model_size=model_size, method=method)
     _ = asyncio.get_event_loop().run_until_complete(
-        reader.method_selection(result_path=result_path, data_path=str(data_path))
+        reader.method_selection(
+            image=cv2.imread(str(data_path)), result_path=result_path
+        )
     )