photoprism/internal/config/tensorflow.go

23 lines
550 B
Go
Raw Normal View History

package config
import (
"path/filepath"
tf "github.com/tensorflow/tensorflow/tensorflow/go"
)
// TensorFlowVersion returns the TenorFlow framework version.
func (c *Config) TensorFlowVersion() string {
return tf.Version()
}
// TensorFlowModelPath returns the TensorFlow model path.
func (c *Config) TensorFlowModelPath() string {
return filepath.Join(c.AssetsPath(), "nasnet")
}
// NSFWModelPath returns the "not safe for work" TensorFlow model path.
func (c *Config) NSFWModelPath() string {
return filepath.Join(c.AssetsPath(), "nsfw")
}