setting numpy and opencv versions, fixing video colormap bug with new…

… versions
bdaiinstitute · Oct 22, 2024 · 3c3f64f · 3c3f64f
1 parent f443d91
commit 3c3f64f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ requires-python = ">=3.9"
 dependencies = [
     "torch == 1.12.1",
     "torchvision == 0.13.1",
-    "numpy >= 1.22.4",
+    "numpy == 1.26.4",
     "flask >= 2.3.2",
     "seaborn >= 0.12.2",  # required by yolov7
     "open3d >= 0.17.0",
@@ -25,6 +25,7 @@ dependencies = [
     "frontier_exploration @ git+https://github.com/naokiyokoyama/frontier_exploration.git",
     "mobile_sam @ git+https://github.com/ChaoningZhang/MobileSAM.git",
     "depth_camera_filtering @ git+https://github.com/naokiyokoyama/depth_camera_filtering",
+    "opencv-python == 4.5.5.64"
 ]
 
 [project.optional-dependencies]

diff --git a/vlfm/vlm/detections.py b/vlfm/vlm/detections.py
@@ -211,7 +211,7 @@ def draw_bounding_box(
 
     if color is None:
         # Randomly choose a color from the rainbow colormap (so boxes aren't black)
-        single_pixel = np.array([[np.random.randint(0, 256)]])
+        single_pixel = np.array([[np.random.randint(0, 256)]], dtype=np.uint8)
         single_pixel = cv2.applyColorMap(single_pixel, cv2.COLORMAP_RAINBOW)
 
         # reshape to a single dimensional array