From 350e8eb69e01bb162ec0b22d1d13a1d1c2752853 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 26 Nov 2022 17:33:43 +0100 Subject: [PATCH] Fix SegmentationModel Usage (#10303) Update export.py Signed-off-by: Glenn Jocher Signed-off-by: Glenn Jocher --- export.py | 1 + 1 file changed, 1 insertion(+) diff --git a/export.py b/export.py index e43d9b7..3028e65 100644 --- a/export.py +++ b/export.py @@ -596,6 +596,7 @@ def run( f = [str(x) for x in f if x] # filter out '' and None if any(f): cls, det, seg = (isinstance(model, x) for x in (ClassificationModel, DetectionModel, SegmentationModel)) # type + det &= not seg # segmentation models inherit from SegmentationModel(DetectionModel) dir = Path('segment' if seg else 'classify' if cls else '') h = '--half' if half else '' # --half FP16 inference arg s = "# WARNING ⚠️ ClassificationModel not yet supported for PyTorch Hub AutoShape inference" if cls else \