28 lines
732 B
Go
28 lines
732 B
Go
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()
|
|
}
|
|
|
|
// TensorFlowOff returns true if TensorFlow should NOT be used for image classification (or anything else).
|
|
func (c *Config) TensorFlowOff() bool {
|
|
return c.params.TensorFlowOff
|
|
}
|
|
|
|
// 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")
|
|
}
|