From daab99015a59adaac710b8a953bbf587b79e3a4c Mon Sep 17 00:00:00 2001 From: Michael Mayer Date: Thu, 6 May 2021 12:50:53 +0200 Subject: [PATCH] TensorFlow: Improve panic log messages #1270 --- internal/classify/tensorflow.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/internal/classify/tensorflow.go b/internal/classify/tensorflow.go index 517aff4aa..8d4a926a8 100644 --- a/internal/classify/tensorflow.go +++ b/internal/classify/tensorflow.go @@ -62,7 +62,7 @@ func (t *TensorFlow) File(filename string) (result Labels, err error) { func (t *TensorFlow) Labels(img []byte) (result Labels, err error) { defer func() { if r := recover(); r != nil { - err = fmt.Errorf("classify: %s (panic)\nstack: %s", r, debug.Stack()) + err = fmt.Errorf("classify: %s (inference panic)\nstack: %s", r, debug.Stack()) } }() @@ -78,7 +78,7 @@ func (t *TensorFlow) Labels(img []byte) (result Labels, err error) { tensor, err := t.createTensor(img, "jpeg") if err != nil { - return nil, fmt.Errorf("classify: %s", err) + return nil, err } // Run inference.