Update detector to Nodelib 0.13.0

Remove icecream import
zauberzeug · Dec 6, 2024 · 5cda389 · 5cda389
1 parent f6d2226
commit 5cda389
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/detector/docker.sh b/detector/docker.sh
@@ -28,8 +28,8 @@ fi
 
 # ========================== BUILD CONFIGURATION / IMAGE SELECTION =======================
 
-SEMANTIC_VERSION=0.1.8
-NODE_LIB_VERSION=0.11.1
+SEMANTIC_VERSION=0.1.9
+NODE_LIB_VERSION=0.13.0
 build_args=" --build-arg NODE_LIB_VERSION=$NODE_LIB_VERSION"
 
 if [ -f /etc/nv_tegra_release ] # Check if we are on a Jetson device

diff --git a/detector/main.py b/detector/main.py
@@ -1,13 +1,11 @@
 # import logging
 import os
 
-import icecream
 import uvicorn
 from learning_loop_node import DetectorNode
 
 from yolov5_detector import Yolov5Detector
 
-icecream.install()
 reload = os.getenv("UVICORN_RELOAD", "FALSE").lower() == "true"
 print(f'Uvicorn reload is set to: {reload}')
 

diff --git a/detector/yolov5_detector.py b/detector/yolov5_detector.py
@@ -8,9 +8,10 @@
 
 import cv2
 import numpy as np
-from learning_loop_node.data_classes import (BoxDetection, CategoryType,
-                                             ImageMetadata, PointDetection)
+from learning_loop_node.data_classes import (BoxDetection, ImageMetadata,
+                                             PointDetection)
 from learning_loop_node.detector.detector_logic import DetectorLogic
+from learning_loop_node.enums import CategoryType
 
 import yolov5