From 8473e4eb6286e53401494642e4636c1499d0cf42 Mon Sep 17 00:00:00 2001 From: Samuel Dobron Date: Thu, 27 Jun 2024 13:56:38 +0000 Subject: [PATCH] exporter: 16bit single-channel network support --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 3f3a91a..eb1e831 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -176,7 +176,7 @@ class Exporter: raise SystemError('Edge TPU export only supported on Linux. See https://coral.ai/docs/edgetpu/compiler/') # Input - im = torch.zeros(self.args.batch, 3, *self.imgsz).to(self.device) + im = torch.zeros(self.args.batch, 1, *self.imgsz).to(self.device) file = Path( getattr(model, 'pt_path', None) or getattr(model, 'yaml_file', None) or model.yaml.get('yaml_file', '')) if file.suffix in ('.yaml', '.yml'):