Conv

Imageomics · Feb 16, 2025 · 387d083 · 387d083
1 parent 5d1b073
commit 387d083
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,7 @@ authors = [
 ]
 description = "Tools for working with data for annotating animal behavior. These were specifically designed during construction of the KABR dataset."
 readme = "README.md"
-requires-python = ">=3.10, <3.11"
+requires-python = ">=3.10, <3.12"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: MIT License",

diff --git a/src/kabr_tools/utils/yolo.py b/src/kabr_tools/utils/yolo.py
@@ -4,10 +4,11 @@
 from torch.nn.modules.container import Sequential
 from ultralytics import YOLO
 from ultralytics.nn.tasks import DetectionModel
+from ultralytics.nn.modules import Conv
 
 torch.serialization.add_safe_globals([Sequential])
 torch.serialization.add_safe_globals([DetectionModel])
-
+torch.serialization.add_safe_globals([Conv])
 
 class YOLOv8:
     def __init__(self, weights="yolov8x.pt", imgsz=640, conf=0.5):