From cefb09e5c8d6c1c303201a6e19ba7fc3fcc77e26 Mon Sep 17 00:00:00 2001 From: Nick White Date: Sun, 18 Dec 2022 15:16:27 +1000 Subject: [PATCH] Fixes #1160 Allow SAHI models to load for colab / intro tutorials --- icevision/models/inference_sahi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/icevision/models/inference_sahi.py b/icevision/models/inference_sahi.py index bd13ec561..22080c73d 100644 --- a/icevision/models/inference_sahi.py +++ b/icevision/models/inference_sahi.py @@ -11,7 +11,7 @@ from icevision.tfms.albumentations import albumentations_adapter -from sahi.models.base import DetectionModel +from sahi import DetectionModel from sahi.prediction import ObjectPrediction from sahi.predict import get_sliced_prediction as sahi_get_sliced_prediction