initial-commit

main.py

+import os
+from dotenv import load_dotenv
+
+from src.scripts.pipe_counting import perform_infer
+
+load_dotenv()
+
+image_path = os.getenv('IMAGE_PATH',"")
+model_path = os.getenv('MODEL_PATH',"model.pt")
+
+if __name__ == '__main__':
+    print(image_path, model_path)
+    perform_infer(image_path=image_path, model_path=model_path)

src/scripts/pipe_counting.py

+from yolov5.detect import run
+from yolov5.detect import import load_model
+
+def load_model(model_path: str):
+    return load_model(weights=model_path)
+
+def perform_infer(image_path: str, model_path: str):
+    return run(source=image_path)
\ No newline at end of file

yolov5 @ 02b8a4c2

+Subproject commit 02b8a4c21bb6d9419bbf01d4af20724743dab58b