luisgulo/photoprism
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

TensorFlow: Improve panic log messages #1270

Browse Source
This commit is contained in:
Michael Mayer 2021-05-06 12:50:53 +02:00
parent cd0f201ef4
commit daab99015a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/classify/tensorflow.go
View File

@ -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.