Conv

Author: zhong-al

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",

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):